model/model_0/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": 10, "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.6981074272290521, 0.2594577281371407, 0.16056970701269482, 0.12656819302102792, 0.10815351424009903, 0.10907206441397252, 0.08555250760653745, 0.07441781347860461, 0.07726245827001074, 0.0703853020525497], "EPOCHS_DONE": 10, "TRAINED_MODEL_DIR": "/Users/suyash.harlalka/Desktop/personal/acne_classification/model/model_0"}