Messis | W&B Run exp-13-neck-reduction-bottleneck-rf-4-2 (https://wandb.ai/crop-classification/messis/runs/65fnb0ca)

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,14 +10,15 @@
       4,
       5
     ],
-    "batch_size": 4,
+    "batch_size": 8,
+    "bottleneck_reduction_factor": 4,
     "debug": false,
     "dropout_p": 0.1,
     "early_stopping_metric": "val_f1_tier3_majority",
     "early_stopping_mode": "max",
     "early_stopping_patience": 10,
-    "experiment_group": "exp-9-dropout2d",
-    "experiment_name": "exp-9-dropout2d-p-0.1",
+    "experiment_group": "exp-13-neck-reduction",
+    "experiment_name": "exp-13-neck-reduction-bottleneck-rf-4",
     "freeze_backbone": true,
     "heads_spec": {
       "tier1": {
@@ -54,7 +55,7 @@
     "img_size": 224,
     "lr": 0.001,
     "max_epochs": 400,
-    "name": "p-0.1",
+    "name": "bottleneck-rf-4",
     "num_frames": 3,
     "optimizer": "Adam",
     "optimizer_momentum": null,
@@ -65,12 +66,13 @@
     ],
     "train_folds": [
       0,
-      2,
+      1,
       3,
       4
     ],
+    "use_bottleneck_neck": true,
     "val_folds": [
-      1
+      2
     ]
   }
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:29fcf50e473cb9ad9b62c187aeac52e2cd07d6e5f5fb5cbe4d1d629b15debafc
-size 713880940
+oid sha256:c7911abf2d63b687dd4fd72109f12f9d73a1bf71b32494702c8493977d750136
+size 416634516