Messis | W&B Run exp-3-timeranges-april-august-4 (https://wandb.ai/crop-classification/messis/runs/7vi3maq1)

config.json

@@ -1,76 +1,56 @@
-{
-  "hparams": {
-    "accumulate_grad_batches": 1,
-    "backbone_weights_path": "./prithvi/models/Prithvi_100M.pt",
-    "bands": [
-      0,
-      1,
-      2,
-      3,
-      4,
-      5
-    ],
-    "batch_size": 16,
-    "debug": false,
-    "dropout_p": 0.1,
-    "early_stopping_metric": "val_f1_tier3_majority",
-    "early_stopping_mode": "max",
-    "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.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
-    ],
-    "val_folds": [
-      4
-    ]
-  }
+{
+  "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
+    ]
+  }
 }

model.safetensors

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