Spaces:
Runtime error
Runtime error
Tanishq commited on
Commit ·
f248f7a
1
Parent(s): 99aa1b0
Update config.py
Browse files
config.py
CHANGED
|
@@ -1,7 +1,7 @@
|
|
| 1 |
import albumentations as A
|
| 2 |
from albumentations.pytorch import ToTensorV2
|
| 3 |
|
| 4 |
-
DEVICE = "
|
| 5 |
TRAIN_DIR = "data/train"
|
| 6 |
VAL_DIR = "data/val"
|
| 7 |
LEARNING_RATE = 2e-4
|
|
|
|
| 1 |
import albumentations as A
|
| 2 |
from albumentations.pytorch import ToTensorV2
|
| 3 |
|
| 4 |
+
DEVICE = "cpu"
|
| 5 |
TRAIN_DIR = "data/train"
|
| 6 |
VAL_DIR = "data/val"
|
| 7 |
LEARNING_RATE = 2e-4
|