Update resnet_lightning.py

resnet_lightning.py

@@ -46,7 +46,7 @@ class BasicBlock(nn.Module):
 
 
 class ResNet18Model(L.LightningModule):
-    def __init__(self, data_dir=PATH_DATASETS, block=BasicBlock, num_blocks=[2, 2, 2, 2], num_classes=10):
+    def __init__(self, data_dir="./data", block=BasicBlock, num_blocks=[2, 2, 2, 2], num_classes=10):
         super(ResNet18Model, self).__init__()
         self.data_dir = data_dir
         self.num_classes = num_classes