ibrahimdaud commited on
Commit
d2d7bb5
·
verified ·
1 Parent(s): fc46c81

Upload multilabel_freeze_metadata.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. multilabel_freeze_metadata.json +43 -0
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
+ }