Fix typo

Former-commit-id: 72f12645410e87ba18093f055c9d8d9f8045595b
2019-10-30 12:27:03 +01:00 · 2019-10-30 12:27:03 +01:00 · 6f23624412
parent 4c0f0a7a7b
commit 6f23624412
1 changed files with 1 additions and 1 deletions
--- a/predict.py
+++ b/predict.py
@ -111,7 +111,7 @@ if __name__ == "__main__":

    device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
    logging.info(f'Using device {device}')
-    net.to(deviec=device)
+    net.to(device=device)
    net.load_state_dict(torch.load(args.model, map_location=device))

    logging.info("Model loaded !")