oskarastrom commited on
Commit
77f857d
·
1 Parent(s): f933454

Update track_detection.py

Browse files
Files changed (1) hide show
  1. scripts/track_detection.py +1 -1
scripts/track_detection.py CHANGED
@@ -79,7 +79,7 @@ def track(in_loc_dir, out_loc_dir, metadata_path, seq, config, verbose):
79
  out_path = os.path.join(out_loc_dir, seq + ".txt")
80
 
81
 
82
- if torch.cude.is_available():
83
  device_name = 'cuda:0'
84
  device = torch.device(device_name)
85
  inference = torch.load(inference_path, map_location=device)
 
79
  out_path = os.path.join(out_loc_dir, seq + ".txt")
80
 
81
 
82
+ if torch.cuda.is_available():
83
  device_name = 'cuda:0'
84
  device = torch.device(device_name)
85
  inference = torch.load(inference_path, map_location=device)