FILTER_COUNT = 32
KERNEL_SIZE = 2
CROP_SIZE = 224
BATCH_SIZE = 256
EPOCHS = 20