Messis | W&B Run exp-4-backbone-frozen-1 (https://wandb.ai/crop-classification/messis/runs/pcifa7t0)

config.json

@@ -1,6 +1,6 @@
 {
   "hparams": {
-    "accumulate_grad_batches": 4,
+    "accumulate_grad_batches": 1,
     "backbone_weights_path": "./prithvi/models/Prithvi_100M.pt",
     "bands": [
       0,
@@ -10,52 +10,67 @@
       4,
       5
     ],
-    "batch_size": 4,
+    "batch_size": 16,
     "debug": false,
-    "early_stopping_metric": "val_loss",
-    "early_stopping_mode": "min",
+    "dropout_p": 0.1,
+    "early_stopping_metric": "val_f1_tier3_majority",
+    "early_stopping_mode": "max",
     "early_stopping_patience": 10,
-    "experiment_group": "exp-2-optimizer-lr",
-    "experiment_name": "exp-2-optimizer-lr-sgd-momentum",
+    "experiment_group": "exp-4-backbone",
+    "experiment_name": "exp-4-backbone-frozen",
     "freeze_backbone": true,
-    "img_size": 224,
-    "lr": 0.001,
-    "max_epochs": 400,
-    "name": "sgd-momentum",
-    "num_frames": 3,
-    "optimizer": "SGD",
-    "optimizer_momentum": 0.9,
-    "optimizer_weight_decay": null,
-    "patch_size": 16,
-    "test_folds": [
-      5
-    ],
-    "tiers": {
+    "heads_spec": {
       "tier1": {
+        "is_metrics_tier": true,
         "loss_weight": 1,
-        "num_classes": 6
+        "num_classes_to_predict": 6,
+        "target_idx": 0,
+        "type": "HierarchicalFCNHead"
       },
       "tier2": {
+        "is_metrics_tier": true,
         "loss_weight": 1,
-        "num_classes": 17
+        "num_classes_to_predict": 17,
+        "target_idx": 1,
+        "type": "HierarchicalFCNHead"
       },
       "tier3": {
+        "is_last_tier": true,
+        "is_metrics_tier": true,
         "loss_weight": 1,
-        "num_classes": 49
+        "num_classes_to_predict": 49,
+        "target_idx": 2,
+        "type": "HierarchicalFCNHead"
       },
-      "tier3_refined": {
+      "tier3_refinement_head": {
+        "is_final_head": true,
+        "is_metrics_tier": false,
         "loss_weight": 1,
-        "num_classes": 49
+        "num_classes_to_predict": 49,
+        "target_idx": 2,
+        "type": "LabelRefinementHead"
       }
     },
+    "img_size": 224,
+    "lr": 0.001,
+    "max_epochs": 400,
+    "name": "frozen",
+    "num_frames": 3,
+    "optimizer": "Adam",
+    "optimizer_momentum": null,
+    "optimizer_weight_decay": null,
+    "patch_size": 16,
+    "test_folds": [
+      5
+    ],
     "train_folds": [
       0,
-      1,
       2,
-      3
+      3,
+      4
     ],
     "val_folds": [
-      4
+      1
     ]
   }
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:cb1e240ff3eb9796d9294fafcd9e7328f9bc71a4adac797b92726097137f600a
-size 713881052
+oid sha256:5b02a3d23a67ea106058c8e1ecc462f7eba98c22a49f9f91c8b366597cec9f21
+size 713880940