Spaces:
Runtime error
Runtime error
extra
Browse files
config.py
CHANGED
|
@@ -11,7 +11,7 @@ DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
|
|
| 11 |
NUM_WORKERS = 4
|
| 12 |
BATCH_SIZE = 16
|
| 13 |
IMAGE_SIZE = 416
|
| 14 |
-
NUM_CLASSES =
|
| 15 |
LEARNING_RATE = 1e-5
|
| 16 |
WEIGHT_DECAY = 1e-4
|
| 17 |
NUM_EPOCHS = 100
|
|
|
|
| 11 |
NUM_WORKERS = 4
|
| 12 |
BATCH_SIZE = 16
|
| 13 |
IMAGE_SIZE = 416
|
| 14 |
+
NUM_CLASSES = 20
|
| 15 |
LEARNING_RATE = 1e-5
|
| 16 |
WEIGHT_DECAY = 1e-4
|
| 17 |
NUM_EPOCHS = 100
|