adithiyyha commited on
Commit
2e62993
·
verified ·
1 Parent(s): d832061

Update config.py

Browse files
Files changed (1) hide show
  1. config.py +1 -1
config.py CHANGED
@@ -4,7 +4,7 @@ import torch
4
  from albumentations.pytorch import ToTensorV2
5
 
6
 
7
- CHECKPOINT_FILE = './checkpoints/x_ray_model.pth.tar'
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