Glenn Jocher
4 vuotta sitten
1 muutettua tiedostoa jossa 0 lisäystä ja 2 poistoa
+ 0
- 2
test.py
Näytä tiedosto
| @@ -33,8 +33,6 @@ def test(data, | |||
| google_utils.attempt_download(weights) | |||
| model = torch.load(weights, map_location=device)['model'] | |||
| torch_utils.model_info(model) | |||
| # Fuse | |||
| # model.fuse() | |||
| model.to(device) | |||
Loading…