Update anomaly_detection_pytorch.py

kashif · Nov 5, 2024 · d70e1ed · d70e1ed
1 parent 8b10b66
commit d70e1ed
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/anomaly_detection_pytorch.py b/examples/anomaly_detection_pytorch.py
@@ -115,7 +115,7 @@ def main(args):
 
     anomalies = []
     means = []
-    model.eval()
+    model.cpu().eval()
     with torch.no_grad():
         for batch in tqdm(test_data_loader, desc="Processing batches"):
             inputs = select(