Glenn Jocher
4 anni fa
1 ha cambiato i file con 0 aggiunte e 2 eliminazioni
+ 0
- 2
test.py
Vedi File
| @@ -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…