Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates

This commit is contained in:
Glenn Jocher 2018-12-24 14:33:05 +01:00
parent 5403581e38
commit b6ff9cad79
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
models.py
View File

@ -89,7 +89,7 @@ class Upsample(torch.nn.Module):
self.mode = mode self.mode = mode
def forward(self, x): def forward(self, x):
return torch.nn.functional.interpolate(x, scale_factor=self.scale_factor, mode=self.mode) return nn.functional.interpolate(x, scale_factor=self.scale_factor, mode=self.mode)
class YOLOLayer(nn.Module): class YOLOLayer(nn.Module):