Messis | W&B Run exp-1-dataset-split-strategy-sechidis-0 (https://wandb.ai/crop-classification/messis/runs/hugmrw3l)

config.json

@@ -1,6 +1,6 @@
 {
   "hparams": {
-    "accumulate_grad_batches": 4,
+    "accumulate_grad_batches": 1,
     "backbone_weights_path": "./prithvi/models/Prithvi_100M.pt",
     "bands": [
       0,
@@ -10,15 +10,20 @@
       4,
       5
     ],
-    "batch_size": 4,
+    "batch_size": 16,
     "bottleneck_reduction_factor": 8,
+    "data_augmentation": {
+      "enabled": false,
+      "flip_prob": 0.5,
+      "jitter_std": 0
+    },
     "debug": false,
     "dropout_p": 0.1,
     "early_stopping_metric": "val_f1_tier3_majority",
     "early_stopping_mode": "max",
     "early_stopping_patience": 10,
-    "experiment_group": "exp-12-num_convs",
-    "experiment_name": "exp-12-num_convs-1",
+    "experiment_group": "exp-1-dataset-split-strategy",
+    "experiment_name": "exp-1-dataset-split-strategy-sechidis",
     "freeze_backbone": true,
     "heads_spec": {
       "tier1": {
@@ -26,7 +31,6 @@
         "kernel_size": 3,
         "loss_weight": 1,
         "num_classes_to_predict": 6,
-        "num_convs": 1,
         "target_idx": 0,
         "type": "HierarchicalFCNHead"
       },
@@ -35,7 +39,6 @@
         "kernel_size": 3,
         "loss_weight": 1,
         "num_classes_to_predict": 17,
-        "num_convs": 1,
         "target_idx": 1,
         "type": "HierarchicalFCNHead"
       },
@@ -45,7 +48,6 @@
         "kernel_size": 3,
         "loss_weight": 1,
         "num_classes_to_predict": 49,
-        "num_convs": 1,
         "target_idx": 2,
         "type": "HierarchicalFCNHead"
       },
@@ -59,9 +61,10 @@
       }
     },
     "img_size": 224,
+    "loss_ignore_background": false,
     "lr": 0.001,
     "max_epochs": 400,
-    "name": "1",
+    "name": "sechidis",
     "num_frames": 3,
     "optimizer": "Adam",
     "optimizer_momentum": null,

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:227a1579466b04a8e88ce37595893ff55c91b18eda7cea51efbc5895592b3ed6
+oid sha256:b4bb8ca96b7ed74bc57ae32981cea76810b4e62333d3d55013f9691d30b31226
 size 713880940