Update resnet.py

resnet.py

@@ -24,6 +24,10 @@ from torchvision.datasets import CIFAR10
 from pytorch_lightning.callbacks import ModelCheckpoint, ModelSummary
 import torchvision.transforms as transforms
 
+PATH_DATASETS = os.environ.get("PATH_DATASETS", ".")
+AVAIL_GPUS = min(1, torch.cuda.device_count())
+BATCH_SIZE = 256 if AVAIL_GPUS else 64
+
 
 class BasicBlock(nn.Module):
     expansion = 1