Update track_detection.py

scripts/track_detection.py

@@ -79,7 +79,7 @@ def track(in_loc_dir, out_loc_dir, metadata_path, seq, config, verbose):
 
     device_name = '0' if torch.cuda.is_available() else 'cpu'
     device = select_device(device_name, batch_size=32)
-    inference = torch.load(inference_path, device=device)
+    inference = torch.load(inference_path, map_location=device)
 
     print(type(inference))
     print(len(inference))