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

config.json

@@ -1,56 +1,76 @@
-{
-  "hparams": {
-    "accumulate_grad_batches": 8,
-    "backbone_weights_path": "./prithvi/models/Prithvi_100M.pt",
-    "bands": [
-      0,
-      1,
-      2,
-      3,
-      4,
-      5
-    ],
-    "batch_size": 2,
-    "debug": false,
-    "early_stopping_patience": 10,
-    "experiment_group": "exp-3-timeranges",
-    "experiment_name": "exp-3-timeranges-april-august",
-    "freeze_backbone": true,
-    "img_size": 224,
-    "lr": 0.001,
-    "max_epochs": 100,
-    "name": "april-august",
-    "num_frames": 3,
-    "patch_size": 16,
-    "test_folds": [
-      5
-    ],
-    "tiers": {
-      "tier1": {
-        "loss_weight": 1,
-        "num_classes": 6
-      },
-      "tier2": {
-        "loss_weight": 1,
-        "num_classes": 17
-      },
-      "tier3": {
-        "loss_weight": 1,
-        "num_classes": 49
-      },
-      "tier3_refined": {
-        "loss_weight": 1,
-        "num_classes": 49
-      }
-    },
-    "train_folds": [
-      0,
-      1,
-      2,
-      3
-    ],
-    "val_folds": [
-      4
-    ]
-  }
+{
+  "hparams": {
+    "accumulate_grad_batches": 4,
+    "backbone_weights_path": "./prithvi/models/Prithvi_100M.pt",
+    "bands": [
+      0,
+      1,
+      2,
+      3,
+      4,
+      5
+    ],
+    "batch_size": 4,
+    "debug": false,
+    "dropout_p": 0.1,
+    "early_stopping_metric": "val_loss",
+    "early_stopping_mode": "min",
+    "early_stopping_patience": 10,
+    "experiment_group": "exp-4-backbone",
+    "experiment_name": "exp-4-backbone-frozen",
+    "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,
+        "num_classes_to_predict": 49,
+        "target_idx": 2,
+        "type": "HierarchicalFCNHead"
+      },
+      "tier3_refinement_head": {
+        "is_final_head": true,
+        "is_metrics_tier": false,
+        "loss_weight": 1,
+        "num_classes_to_predict": 49,
+        "target_idx": 2,
+        "type": "LabelRefinementHead"
+      }
+    },
+    "img_size": 224,
+    "lr": 0.0001,
+    "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": [
+      1,
+      2,
+      3,
+      4
+    ],
+    "val_folds": [
+      0
+    ]
+  }
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:30a40106a0be735f9330b5cafcc368e921263374c4612331898e326799a400c6
-size 713881052
+oid sha256:d25b9a2cb124571c34ef9bcf719ca9df5696438d3226fee62d9fdc9f7b42f95d
+size 713880940