Messis | W&B Run exp-8-drop-hierarchy-full_hierarchy-0 (https://wandb.ai/crop-classification/messis/runs/x6fjdbh6)

config.json

@@ -1,7 +1,7 @@
 {
   "hparams": {
-    "accumulate_grad_batches": 1,
-    "backbone_weights_path": null,
+    "accumulate_grad_batches": 4,
+    "backbone_weights_path": "./prithvi/models/Prithvi_100M.pt",
     "bands": [
       0,
       1,
@@ -10,15 +10,15 @@
       4,
       5
     ],
-    "batch_size": 16,
+    "batch_size": 4,
     "debug": false,
     "dropout_p": 0.1,
-    "early_stopping_metric": "val_loss",
-    "early_stopping_mode": "min",
+    "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-not_pretrained-vles",
-    "freeze_backbone": false,
+    "experiment_group": "exp-8-drop-hierarchy",
+    "experiment_name": "exp-8-drop-hierarchy-full_hierarchy",
+    "freeze_backbone": true,
     "heads_spec": {
       "tier1": {
         "is_metrics_tier": true,
@@ -52,9 +52,9 @@
       }
     },
     "img_size": 224,
-    "lr": 0.001,
+    "lr": 0.0001,
     "max_epochs": 400,
-    "name": "not_pretrained-vles",
+    "name": "full_hierarchy",
     "num_frames": 3,
     "optimizer": "Adam",
     "optimizer_momentum": null,
@@ -64,13 +64,13 @@
       5
     ],
     "train_folds": [
-      0,
       1,
       2,
-      3
+      3,
+      4
     ],
     "val_folds": [
-      4
+      0
     ]
   }
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:2bfd2b5dab8739137943025f66989c5d86ca4947b96df6433819b8a723e8a807
+oid sha256:615385b90edc5e3f67f8870f8b1fa230bfac4dbf5865af657d6251b35c5af206
 size 713880940