mirrors/yolov5
1
0
Fork 0
mirror of https://github.com/ultralytics/yolov5.git synced 2025-06-03 14:49:29 +08:00
Code Issues Projects Releases Wiki Activity

yolo.py bug fix

Browse Source
This commit is contained in:
Glenn Jocher 2020-07-13 15:21:19 -07:00
parent 02445d176d
commit 2b1892430f
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/yolo.py
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):