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