Messis | W&B Run exp-7-label-hierarchy-crop-seasonality-1 (https://wandb.ai/crop-classification/messis/runs/13q9u9rd)

config.json

@@ -1,6 +1,6 @@
 {
   "hparams": {
-    "accumulate_grad_batches": 16,
+    "accumulate_grad_batches": 4,
     "backbone_weights_path": "./prithvi/models/Prithvi_100M.pt",
     "bands": [
       0,
@@ -10,7 +10,7 @@
       4,
       5
     ],
-    "batch_size": 1,
+    "batch_size": 4,
     "bottleneck_reduction_factor": 8,
     "data_augmentation": {
       "enabled": false,
@@ -19,44 +19,34 @@
     },
     "debug": false,
     "dropout_p": 0.1,
-    "early_stopping_metric": "val_f1_tier3_majority",
+    "early_stopping_metric": "val_f1_tier2_majority",
     "early_stopping_mode": "max",
     "early_stopping_patience": 10,
-    "experiment_group": "exp-14-more-timesteps",
-    "experiment_name": "exp-14-more-timesteps-9",
+    "experiment_group": "exp-7-label-hierarchy",
+    "experiment_name": "exp-7-label-hierarchy-crop-seasonality",
     "freeze_backbone": true,
     "heads_spec": {
       "tier1": {
         "is_metrics_tier": true,
-        "kernel_size": 3,
         "loss_weight": 1,
-        "num_classes_to_predict": 6,
+        "num_classes_to_predict": 4,
         "target_idx": 0,
         "type": "HierarchicalFCNHead"
       },
       "tier2": {
-        "is_metrics_tier": true,
-        "kernel_size": 3,
-        "loss_weight": 1,
-        "num_classes_to_predict": 17,
-        "target_idx": 1,
-        "type": "HierarchicalFCNHead"
-      },
-      "tier3": {
         "is_last_tier": true,
         "is_metrics_tier": true,
-        "kernel_size": 3,
         "loss_weight": 1,
-        "num_classes_to_predict": 49,
-        "target_idx": 2,
+        "num_classes_to_predict": 43,
+        "target_idx": 1,
         "type": "HierarchicalFCNHead"
       },
-      "tier3_refinement_head": {
+      "tier2_refinement_head": {
         "is_final_head": true,
         "is_metrics_tier": false,
         "loss_weight": 1,
-        "num_classes_to_predict": 49,
-        "target_idx": 2,
+        "num_classes_to_predict": 43,
+        "target_idx": 1,
         "type": "LabelRefinementHead"
       }
     },
@@ -64,8 +54,8 @@
     "loss_ignore_background": false,
     "lr": 0.001,
     "max_epochs": 400,
-    "name": "9",
-    "num_frames": 9,
+    "name": "crop-seasonality",
+    "num_frames": 3,
     "optimizer": "Adam",
     "optimizer_momentum": null,
     "optimizer_weight_decay": null,
@@ -75,13 +65,13 @@
     ],
     "train_folds": [
       0,
-      1,
       2,
-      3
+      3,
+      4
     ],
     "use_bottleneck_neck": false,
     "val_folds": [
-      4
+      1
     ]
   }
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:4aa6c07203c9804e06c4bc53b82ce87bc024eabd91c1dd95be2d25b248ef64ff
-size 1435975780
+oid sha256:5fe375d8521b4556ef264a4a547dfa2ec35c6b9aec44333075f6189a5d0f0732
+size 711475104