adithiyyha commited on
Commit
6de049b
·
verified ·
1 Parent(s): c486a3c

Update AKSHAYRAJAA/config.py

Browse files
Files changed (1) hide show
  1. 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\\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
 
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