pull/1/head
Glenn Jocher 5 years ago
parent 8fe299f179
commit 546f973562

@ -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…
Cancel
Save