Hugging Face's logo

Spaces:
mkthoma
/
cifar10_customResNet
Sleeping

App Files Community
mkthoma commited on
Commit
83214bf
·
1 Parent(s): 1edae1c

Update resnet.py

Browse files
Files changed (1) hide show
  1. resnet.py +1 -1
resnet.py CHANGED
@@ -25,7 +25,7 @@ import matplotlib.pyplot as plt
25
 
26
 
27
  PATH_DATASETS = os.environ.get("PATH_DATASETS", ".")
28
- BATCH_SIZE = 256 if AVAIL_GPUS else 64
29
 
30
  # Model
31
  class custom_ResNet(pl.LightningModule):
 
25
 
26
 
27
  PATH_DATASETS = os.environ.get("PATH_DATASETS", ".")
28
+ BATCH_SIZE = 256
29
 
30
  # Model
31
  class custom_ResNet(pl.LightningModule):