test(resnet9): move model to cpu for ci

joennlae · Sep 20, 2023 · 3b4b6c8 · 3b4b6c8
1 parent 4e2d88e
commit 3b4b6c8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/python/test/model_eval.py b/src/python/test/model_eval.py
@@ -75,14 +75,14 @@ def download_model(
         dtype=torch.qint8,
     )
 
-    model_int8.to("cuda")
+    model_int8.to("cpu")
     criterion = nn.CrossEntropyLoss()
 
     acc1, acc5, loss = evaluate(
         model_int8,
         criterion=criterion,
         data_loader=data_loader_val,
-        device="cuda",
+        device="cpu",
     )
 
     assert acc1 > 0.90