Spaces:

cycool29
/

Spiral-ND

Running

cycool29 commited on Oct 19, 2023

Commit

8eea308

1 Parent(s): d6fe214

Update configs.py

Files changed (1) hide show

configs.py CHANGED Viewed

@@ -44,8 +44,8 @@ LEARNING_RATE = 0.0001
 STEP_SIZE = 10
 GAMMA = 0.3
 CUTMIX_ALPHA = 0.3
-DEVICE = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
-# DEVICE = torch.device("cpu")
 NUM_PRINT = 100
 TASK = 1
 WARMUP_EPOCHS = 5

 STEP_SIZE = 10
 GAMMA = 0.3
 CUTMIX_ALPHA = 0.3
+# DEVICE = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
+DEVICE = torch.device("cpu")
 NUM_PRINT = 100
 TASK = 1
 WARMUP_EPOCHS = 5