Ankle/yolov5_birdproject
1
0 Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

yolo.py bug fix

Browse Source
This commit is contained in:
Glenn Jocher committed 2020-07-13 15:21:19 -07:00
1 parent 02445d176d
commit 2b1892430f
1 file changed
+1 -2
Download Patch File Download Diff File Expand all files Collapse all files
models/yolo.py
+1 -2
View File
@@ -74,8 +74,7 @@ class Model(nn.Module):
# Init weights, biases # Init weights, biases
torch_utils.initialize_weights(self) torch_utils.initialize_weights(self)
self._initialize_biases() # only run once self.info()
torch_utils.model_info(self)
print('') print('')
def forward(self, x, augment=False, profile=False): def forward(self, x, augment=False, profile=False):