| { | |
| "model_architecture": "efficientnet_b0", | |
| "input_size": [3, 224, 224], | |
| "num_classes": 10, | |
| "dataset": "Custom Dataset", | |
| "loss_function": "CrossEntropyLoss", | |
| "optimizer": "AdamW", | |
| "learning_rate": 0.001, | |
| "epochs": 3, | |
| "batch_size": 32, | |
| "validation_accuracy": 85.3, | |
| "class_to_idx": {"class_0": 0, "class_1": 1, "class_2": 2, "class_3": 3, "class_4": 4, "class_5": 5, "class_6": 6, "class_7": 7, "class_8": 8, "class_9": 9}, | |
| "device": "cuda", | |
| "description": "This is an EfficientNet-B0 model trained on a custom dataset for image classification tasks." | |
| } | |