From 6f23624412867936ca6ef17addb474b1f9d18fef Mon Sep 17 00:00:00 2001 From: milesial Date: Wed, 30 Oct 2019 12:27:03 +0100 Subject: [PATCH] Fix typo Former-commit-id: 72f12645410e87ba18093f055c9d8d9f8045595b --- predict.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/predict.py b/predict.py index cf6f1a3..f8a2894 100755 --- 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 !")