model/model_2/config.json

{"DATASET_PATH": "/Users/suyash.harlalka/Desktop/personal/acne_classification/dataset/Classification/JPEGImages", "MODEL_DIR": "/Users/suyash.harlalka/Desktop/personal/acne_classification/model", "MODEL_OBJECTIVE": "REGRESSION", "MODEL_BACKBONE": "EFFICIENT_NET_B0", "MODEL_INPUT_SIZE": [224, 224], "TRANSFORMS_TO_APPLY": ["RESIZE", "TO_TENSOR", "RANDOM_HORIZONTAL_FLIP", "NORMALIZE"], "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, "LOSS": [0.7182596655268418, 0.28382997606929977, 0.21820551078570516, 0.16277591256718887, 0.11228116758559879, 0.11437285848354038, 0.11287243585837514, 0.10567777721505416], "EPOCHS_DONE": 8, "TRAINED_MODEL_DIR": "/Users/suyash.harlalka/Desktop/personal/acne_classification/model/model_2"}