<filename>src/config.py
DEVICE = 'cuda'
LAMBDA = 0.0051
IMG_SIZE = 224
BATCH_SIZE = 256
LR = 0.1
WEIGHT_DECAY = 1e-6
EPOCHS = 100

<filename>src/config.py
DEVICE = 'cuda'
LAMBDA = 0.0051
IMG_SIZE = 224
BATCH_SIZE = 256
LR = 0.1
WEIGHT_DECAY = 1e-6
EPOCHS = 100
