Disable cpu for cifar

pomonam · Mar 20, 2024 · da7fc14 · da7fc14
1 parent 38bdb08
commit da7fc14
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/cifar/analyze.py b/examples/cifar/analyze.py
@@ -115,7 +115,7 @@ def main():
         analysis_name=args.dataset_name,
         model=model,
         task=task,
-        cpu=True,
+        cpu=False,
     )
 
     factor_args = FactorArguments(strategy=args.factor_strategy)