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

updates

This commit is contained in:
Glenn Jocher 2020-01-23 15:15:53 -08:00
parent 07c8a03aa0
commit d498193456
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
models.py
View File

@ -222,8 +222,7 @@ class Darknet(nn.Module):
def forward(self, x, var=None): def forward(self, x, var=None):
img_size = x.shape[-2:] img_size = x.shape[-2:]
layer_outputs = [] output, layer_outputs = [], []
output = []
verbose = False verbose = False
if verbose: if verbose:
print('0', x.shape) print('0', x.shape)