Messis | W&B Run exp-8-drop-hierarchy-no_head-0 (https://wandb.ai/crop-classification/messis/runs/volv8pvr)

config.json

@@ -1,6 +1,6 @@
 {
   "hparams": {
-    "accumulate_grad_batches": 1,
+    "accumulate_grad_batches": 4,
     "backbone_weights_path": "./prithvi/models/Prithvi_100M.pt",
     "bands": [
       0,
@@ -10,34 +10,20 @@
       4,
       5
     ],
-    "batch_size": 16,
-    "debug": true,
+    "batch_size": 4,
+    "debug": false,
     "dropout_p": 0.1,
     "early_stopping_metric": "val_f1_tier3_majority",
     "early_stopping_mode": "max",
     "early_stopping_patience": 10,
-    "experiment_group": "exp-x-accuracy-test",
-    "experiment_name": "exp-x-accuracy-test-frozen",
+    "experiment_group": "exp-8-drop-hierarchy",
+    "experiment_name": "exp-8-drop-hierarchy-no_head",
     "freeze_backbone": true,
     "heads_spec": {
-      "tier1": {
-        "is_metrics_tier": true,
-        "loss_weight": 1,
-        "num_classes_to_predict": 6,
-        "target_idx": 0,
-        "type": "HierarchicalFCNHead"
-      },
-      "tier2": {
-        "is_metrics_tier": true,
-        "loss_weight": 1,
-        "num_classes_to_predict": 17,
-        "target_idx": 1,
-        "type": "HierarchicalFCNHead"
-      },
       "tier3": {
         "is_last_tier": true,
         "is_metrics_tier": true,
-        "loss_weight": 1,
+        "loss_weight": 0,
         "num_classes_to_predict": 49,
         "target_idx": 2,
         "type": "HierarchicalFCNHead"
@@ -52,9 +38,9 @@
       }
     },
     "img_size": 224,
-    "lr": 0.0001,
+    "lr": 0.001,
     "max_epochs": 400,
-    "name": "frozen",
+    "name": "no_head",
     "num_frames": 3,
     "optimizer": "Adam",
     "optimizer_momentum": null,

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:97f20084b93db3e00337c60cc3e68d74b7a68735f761382f9e64032fe95d1920
-size 713880940
+oid sha256:a9200dcd3183db28653d807186ba84061d2952d1d8b7f9ae4d64e028f44a2a78
+size 709117048