Upload checkpoints/cnn_vit_gnn__attention/model_delta_metadata.json with huggingface_hub

checkpoints/cnn_vit_gnn__attention/model_delta_metadata.json

@@ -0,0 +1,129 @@
+{
+  "branches": [
+    "cnn",
+    "vit",
+    "gnn"
+  ],
+  "mode": "attention",
+  "seed": 42,
+  "config": {
+    "root_img": "/kaggle/input/datasets/nilnguyenquoc/mit67-701515/MIT67_701515",
+    "root_graph": "/kaggle/input/datasets/nilnguyenquoc/mit67-graphs-yolov8xworld-k3-c025-m24-vis32-iou30",
+    "out_dir": "/kaggle/working",
+    "hf_upload": true,
+    "hf_upload_repo": "nilnguyen2k/mit67_fusion_head",
+    "hf_upload_private": false,
+    "backbones": {
+      "cnn": {
+        "repo": "nilnguyen2k/mit67_resnet50_places365",
+        "file": "mit_resnet50.pth",
+        "model_name": "resnet50"
+      },
+      "vit": {
+        "repo": "nilnguyen2k/mit67_swin_tiny",
+        "file": "mit67_swin_tiny.pth",
+        "model_name": "swin_tiny_patch4_window7_224"
+      },
+      "gnn": {
+        "repo": "nilnguyen2k/mit67_gnn",
+        "file": "mit67_gnn.pth"
+      }
+    },
+    "num_classes": 67,
+    "imgsz": 224,
+    "extract_batch_size": 64,
+    "fit_split": "valid",
+    "use_pca": true,
+    "pca_dim": 64,
+    "pca_max_dim": 192,
+    "pca_whiten": true,
+    "pca_dim_grid": [
+      32,
+      64,
+      128,
+      192
+    ],
+    "pca_dim_ablation": [
+      192
+    ],
+    "pca_fit_split": "train",
+    "inner_folds": 5,
+    "cv_repeats": 3,
+    "standalone_vit_acc": 87.15,
+    "n_aug_variants": 8,
+    "allow_hflip": false,
+    "dims": {
+      "cnn": 192,
+      "vit": 192,
+      "gnn": 192
+    },
+    "common_dim": 32,
+    "hidden_dim": 0,
+    "dropout_adapter": 0.4,
+    "dropout_head": 0.4,
+    "delta_rank": 12,
+    "delta_l2_penalty": 0.01,
+    "delta_weight_decay": 0.1,
+    "use_delta": true,
+    "epochs_base": 120,
+    "epochs_delta": 30,
+    "epochs": 80,
+    "batch_size": 64,
+    "lr": 0.001,
+    "weight_decay": 0.01,
+    "patience": 0,
+    "mixup_alpha": 0.4,
+    "aux_weight": 0.1,
+    "residual_base_branch": "vit",
+    "delta_scale_init": 1.0,
+    "gate_sup_weight": 0.3,
+    "label_smoothing": 0.1,
+    "log_every": 5,
+    "seeds": [
+      42,
+      1337,
+      2024
+    ],
+    "branch_sets": [
+      [
+        "vit"
+      ],
+      [
+        "cnn",
+        "vit"
+      ],
+      [
+        "vit",
+        "gnn"
+      ],
+      [
+        "cnn",
+        "vit",
+        "gnn"
+      ]
+    ],
+    "modes": [
+      "concat",
+      "attention"
+    ],
+    "gnn_arch": {
+      "vocab_size": 256,
+      "embedding_dim": 32,
+      "box_features": 7,
+      "extra_node_features": 2,
+      "edge_dim": 3,
+      "crop_size": 32,
+      "visual_embed_dim": 16,
+      "hidden_channels": 112,
+      "heads": 4
+    }
+  },
+  "variant": "delta",
+  "metrics": {
+    "accuracy": 87.93677204658901,
+    "precision_macro": 88.33280695473897,
+    "recall_macro": 85.84124321486036,
+    "f1_macro": 86.6769042526632,
+    "f1_weighted": 87.80023336396476
+  }
+}