From ef22a09c5b8322c439767a8321b8e176fa244210 Mon Sep 17 00:00:00 2001 From: Johan Edstedt Date: Sat, 19 Oct 2024 22:59:28 +0200 Subject: [PATCH] cuda default device for eval --- experiments/eval_roma_outdoor.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/experiments/eval_roma_outdoor.py b/experiments/eval_roma_outdoor.py index 7e103be..de549ae 100644 --- 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)