cuda default device for eval

Parskatt · Oct 19, 2024 · ef22a09 · ef22a09
1 parent 0e50b44
commit ef22a09
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/experiments/eval_roma_outdoor.py b/experiments/eval_roma_outdoor.py
@@ -49,7 +49,7 @@ def test_hpatches(model, name):
 
 if __name__ == "__main__":
     from romatch import roma_outdoor
-    device = "cpu"
+    device = "cuda"
     model = roma_outdoor(device = device, coarse_res = 672, upsample_res = 1344)
     experiment_name = "roma_latest"
     test_mega1500(model, experiment_name)