Spaces:

adithiyyha
/

nn

Sleeping

adithiyyha commited on Jan 11

Commit

1bfe09a

verified ·

1 Parent(s): bca1cb3

Update AKSHAYRAJAA/config.py

Files changed (1) hide show

AKSHAYRAJAA/config.py CHANGED Viewed

@@ -4,9 +4,9 @@ import torch
 from albumentations.pytorch import ToTensorV2
-CHECKPOINT_FILE = 'd:\\AKSHAYRAJAA\\checkpoints\\x_ray_model.pth.tar'
-DATASET_PATH = 'D:\\AKSHAYRAJAA\\dataset\\'
-IMAGES_DATASET = 'D:\\AKSHAYRAJAA\\dataset\\images'
 DEVICE = 'cpu'
 BATCH_SIZE = 16

 from albumentations.pytorch import ToTensorV2
+CHECKPOINT_FILE = 'AKSHAYRAJAA\\checkpoints\\x_ray_model.pth.tar'
+DATASET_PATH = 'AKSHAYRAJAA\\dataset\\'
+IMAGES_DATASET = 'AKSHAYRAJAA\\dataset\\images'
 DEVICE = 'cpu'
 BATCH_SIZE = 16