from base import TrainingObjective, ModelBackbone, model_input_dict, TransformationType

DATASET_PATH = '/Users/suyashharlalka/Documents/workspace/gabit/acne_classification/dataset/Classification/JPEGImages'
MODEL_DIR = '/Users/suyashharlalka/Documents/workspace/gabit/acne_classification/model'

MODEL_OBJECTIVE =  TrainingObjective.REGRESSION
MODEL_BACKBONE = ModelBackbone.EFFICIENT_NET_B0
MODEL_INPUT_SIZE = model_input_dict[MODEL_BACKBONE]
TRANSFORMS_TO_APPLY = [
    TransformationType.RESIZE.name, 
    TransformationType.TO_TENSOR.name,
    TransformationType.RANDOM_HORIZONTAL_FLIP.name,
    TransformationType.NORMALIZE.name
]
NUM_CLASSES = 4
LAST_N_LAYERS_TO_TRAIN = 5
EPOCHS = 20
MODEL_TRAINING = True
IS_LIMITED = False
BATCH_SIZE = 64
SHUFFLE = True
NUM_WORKERS = 0

BASE_LR = 0.001
LR_DECAY_STEP_SIZE = 5
LR_DECAY_GAMMA = 0.1