Spaces:
Runtime error
Runtime error
Commit
·
77f857d
1
Parent(s):
f933454
Update track_detection.py
Browse files
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.
|
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)
|