Spaces:
Sleeping
Sleeping
Update AKSHAYRAJAA/config.py
Browse files- AKSHAYRAJAA/config.py +2 -2
AKSHAYRAJAA/config.py
CHANGED
@@ -4,9 +4,9 @@ import torch
|
|
4 |
from albumentations.pytorch import ToTensorV2
|
5 |
|
6 |
|
7 |
-
CHECKPOINT_FILE = 'AKSHAYRAJAA
|
8 |
DATASET_PATH = 'AKSHAYRAJAA/dataset'
|
9 |
-
IMAGES_DATASET = 'AKSHAYRAJAA
|
10 |
|
11 |
DEVICE = 'cpu'
|
12 |
BATCH_SIZE = 16
|
|
|
4 |
from albumentations.pytorch import ToTensorV2
|
5 |
|
6 |
|
7 |
+
CHECKPOINT_FILE = 'AKSHAYRAJAA/checkpoints/x_ray_model.pth.tar'
|
8 |
DATASET_PATH = 'AKSHAYRAJAA/dataset'
|
9 |
+
IMAGES_DATASET = 'AKSHAYRAJAA/dataset/images'
|
10 |
|
11 |
DEVICE = 'cpu'
|
12 |
BATCH_SIZE = 16
|