Pārlūkot izejas kodu

yolo.py bug fix

5.0
Glenn Jocher pirms 4 gadiem
vecāks
revīzija
2b1892430f
1 mainītis faili ar 1 papildinājumiem un 2 dzēšanām
  1. +1
    -2
      models/yolo.py

+ 1
- 2
models/yolo.py Parādīt failu

@@ -74,8 +74,7 @@ class Model(nn.Module):

# Init weights, biases
torch_utils.initialize_weights(self)
self._initialize_biases() # only run once
torch_utils.model_info(self)
self.info()
print('')

def forward(self, x, augment=False, profile=False):

Notiek ielāde…
Atcelt
Saglabāt