Revert "Messis | W&B Run exp-11-more-channels-1024-0 (https://wandb.ai/crop-classification/messis/runs/qyxmbzeo)"

This reverts commit a25b583d440ac44576ad26288b684820aac541fb.

config.json

@@ -1,6 +1,6 @@
 {
   "hparams": {
-    "accumulate_grad_batches": 4,
+    "accumulate_grad_batches": 2,
     "backbone_weights_path": "./prithvi/models/Prithvi_100M.pt",
     "bands": [
       0,
@@ -10,38 +10,50 @@
       4,
       5
     ],
-    "batch_size": 4,
+    "batch_size": 8,
+    "bottleneck_reduction_factor": 4,
+    "data_augmentation": {
+      "enabled": true,
+      "flip_prob": 0.5,
+      "jitter_std": 0.01
+    },
     "debug": false,
     "dropout_p": 0.1,
     "early_stopping_metric": "val_f1_tier3_majority",
     "early_stopping_mode": "max",
     "early_stopping_patience": 10,
-    "experiment_group": "exp-11-more-channels",
-    "experiment_name": "exp-11-more-channels-1024",
-    "freeze_backbone": true,
+    "experiment_group": "best-model",
+    "experiment_name": "best-model-messis",
+    "freeze_backbone": false,
     "heads_spec": {
       "tier1": {
         "is_metrics_tier": true,
-        "kernel_size": 3,
+        "kernel_size": 1,
         "loss_weight": 1,
+        "num_channels": 1024,
         "num_classes_to_predict": 6,
+        "num_convs": 1,
         "target_idx": 0,
         "type": "HierarchicalFCNHead"
       },
       "tier2": {
         "is_metrics_tier": true,
-        "kernel_size": 3,
+        "kernel_size": 1,
         "loss_weight": 1,
+        "num_channels": 1024,
         "num_classes_to_predict": 17,
+        "num_convs": 1,
         "target_idx": 1,
         "type": "HierarchicalFCNHead"
       },
       "tier3": {
         "is_last_tier": true,
         "is_metrics_tier": true,
-        "kernel_size": 3,
+        "kernel_size": 1,
         "loss_weight": 1,
+        "num_channels": 1024,
         "num_classes_to_predict": 49,
+        "num_convs": 1,
         "target_idx": 2,
         "type": "HierarchicalFCNHead"
       },
@@ -55,10 +67,11 @@
       }
     },
     "img_size": 224,
-    "lr": 0.001,
+    "loss_ignore_background": true,
+    "lr": 0.0001,
     "max_epochs": 400,
-    "name": "1024",
-    "num_frames": 3,
+    "name": "messis",
+    "num_frames": 9,
     "optimizer": "Adam",
     "optimizer_momentum": null,
     "optimizer_weight_decay": null,
@@ -67,13 +80,14 @@
       5
     ],
     "train_folds": [
+      0,
       1,
       2,
-      3,
-      4
+      3
     ],
+    "use_bottleneck_neck": false,
     "val_folds": [
-      0
+      4
     ]
   }
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:5beee0006ef65533c4dadd975b382b4bbc73a57bc9a24a74f43c9486482ba403
-size 848618908
+oid sha256:ccfc03917e00e739a26dca031fcabd9692e2038d5c9c48e2c04fe4a87d1ce652
+size 1404514428