Messis | W&B Run best-model-messis-0 (https://wandb.ai/crop-classification/messis/runs/z5nrj5mx)

config.json

@@ -1,6 +1,6 @@
 {
   "hparams": {
-    "accumulate_grad_batches": 8,
+    "accumulate_grad_batches": 2,
     "backbone_weights_path": "./prithvi/models/Prithvi_100M.pt",
     "bands": [
       0,
@@ -10,7 +10,7 @@
       4,
       5
     ],
-    "batch_size": 2,
+    "batch_size": 8,
     "bottleneck_reduction_factor": 4,
     "data_augmentation": {
       "enabled": true,
@@ -21,9 +21,9 @@
     "dropout_p": 0.1,
     "early_stopping_metric": "val_f1_tier3_majority",
     "early_stopping_mode": "max",
-    "early_stopping_patience": 5,
+    "early_stopping_patience": 10,
     "experiment_group": "best-model",
-    "experiment_name": "best-model-messis-test-overfit",
+    "experiment_name": "best-model-messis",
     "freeze_backbone": false,
     "heads_spec": {
       "tier1": {
@@ -70,16 +70,12 @@
     "loss_ignore_background": true,
     "lr": 0.0001,
     "max_epochs": 400,
-    "name": "messis-test-overfit",
+    "name": "messis",
     "num_frames": 9,
     "optimizer": "Adam",
     "optimizer_momentum": null,
     "optimizer_weight_decay": null,
     "patch_size": 16,
-    "subsets": {
-      "train": 4,
-      "val": 2
-    },
     "test_folds": [
       5
     ],
@@ -89,7 +85,7 @@
       3,
       4
     ],
-    "use_bottleneck_neck": true,
+    "use_bottleneck_neck": false,
     "val_folds": [
       0
     ]

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:817890d5f6ffae93f61285b5e9e8d50f6b088d4af57f6fe5b13766a8336d4a41
-size 512775180
+oid sha256:69d173b5bc6d0f38ad5bc7b34d8580c4955b878fddbaac25ca2558dd2d4626f1
+size 1404514428