Upload multilabel_freeze_metadata.json with huggingface_hub
Browse files
multilabel_freeze_metadata.json
ADDED
|
@@ -0,0 +1,43 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"model_type": "multilabel_resnet50",
|
| 3 |
+
"training_mode": "freeze",
|
| 4 |
+
"num_classes": 90,
|
| 5 |
+
"best_map": 0.3746700931482063,
|
| 6 |
+
"epoch": 29,
|
| 7 |
+
"total_parameters": 24656463,
|
| 8 |
+
"config": {
|
| 9 |
+
"experiment_name": "multilabel_freeze_encoder_optimized",
|
| 10 |
+
"training_mode": "freeze",
|
| 11 |
+
"model": {
|
| 12 |
+
"pretrained": true,
|
| 13 |
+
"checkpoint_path": "/content/drive/MyDrive/Colab Notebooks/ai_course_wrok/resnet50_best.pth",
|
| 14 |
+
"dropout": 0.5
|
| 15 |
+
},
|
| 16 |
+
"data": {
|
| 17 |
+
"dataset_name": "ibrahimdaud/multi-label-food-recognition",
|
| 18 |
+
"batch_size": 128,
|
| 19 |
+
"num_workers": 8,
|
| 20 |
+
"image_size": 224,
|
| 21 |
+
"split_ratio": 0.8
|
| 22 |
+
},
|
| 23 |
+
"training": {
|
| 24 |
+
"epochs": 50,
|
| 25 |
+
"learning_rate": 0.001,
|
| 26 |
+
"weight_decay": 0.0001,
|
| 27 |
+
"scheduler": "step",
|
| 28 |
+
"step_size": 10,
|
| 29 |
+
"gamma": 0.1
|
| 30 |
+
},
|
| 31 |
+
"use_amp": true,
|
| 32 |
+
"checkpoint_dir": "/content/drive/MyDrive/Colab Notebooks/ai_course_wrok",
|
| 33 |
+
"resume_checkpoint": null,
|
| 34 |
+
"use_wandb": true,
|
| 35 |
+
"wandb": {
|
| 36 |
+
"project": "multilabel-food-recognition",
|
| 37 |
+
"run_name": "multilabel_freeze_encoder_optimized"
|
| 38 |
+
},
|
| 39 |
+
"debug_mode": false,
|
| 40 |
+
"debug_max_samples": 10,
|
| 41 |
+
"seed": 42
|
| 42 |
+
}
|
| 43 |
+
}
|