Glenn Jocher
преди 4 години
променени са 1 файла, в които са добавени 1 реда и са изтрити 2 реда
+ 1
- 2
models/yolo.py
Целия файл
| @@ -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): | |||
Loading…