params.yaml

IMAGE_SIZE: [150, 150, 3]
BATCH_SIZE: 32
INCLUDE_TOP: False
EPOCHS: 15
CLASSES: 4
WEIGHTS: imagenet