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

README.md

@@ -4,6 +4,6 @@ tags:
 - model_hub_mixin
 ---
 
-This model has been pushed to the Hub using ****:
-- Repo: [More Information Needed]
+This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration:
+- Library: [More Information Needed]
 - Docs: [More Information Needed]

config.json

@@ -1,40 +1,53 @@
-{
-  "hparams": {
-    "accumulate_grad_batches": 1,
-    "backbone_weights_path": "./prithvi/models/Prithvi_100M.pt",
-    "bands": [
-      0,
-      1,
-      2,
-      3,
-      4,
-      5
-    ],
-    "batch_size": 4,
-    "debug": false,
-    "freeze_backbone": false,
-    "img_size": 224,
-    "lr": 0.001,
-    "max_epochs": 1,
-    "num_frames": 3,
-    "patch_size": 16,
-    "tiers": {
-      "tier1": {
-        "loss_weight": 1,
-        "num_classes": 6
-      },
-      "tier2": {
-        "loss_weight": 1,
-        "num_classes": 17
-      },
-      "tier3": {
-        "loss_weight": 1,
-        "num_classes": 49
-      },
-      "tier3_refined": {
-        "loss_weight": 1,
-        "num_classes": 49
-      }
-    }
-  }
+{
+  "hparams": {
+    "accumulate_grad_batches": 8,
+    "backbone_weights_path": "./prithvi/models/Prithvi_100M.pt",
+    "bands": [
+      0,
+      1,
+      2,
+      3,
+      4,
+      5
+    ],
+    "batch_size": 2,
+    "debug": false,
+    "early_stopping_patience": 3,
+    "freeze_backbone": true,
+    "img_size": 224,
+    "lr": 0.001,
+    "max_epochs": 100,
+    "num_frames": 3,
+    "patch_size": 16,
+    "test_folds": [
+      5
+    ],
+    "tiers": {
+      "tier1": {
+        "loss_weight": 1,
+        "num_classes": 6
+      },
+      "tier2": {
+        "loss_weight": 1,
+        "num_classes": 17
+      },
+      "tier3": {
+        "loss_weight": 1,
+        "num_classes": 49
+      },
+      "tier3_refined": {
+        "loss_weight": 1,
+        "num_classes": 49
+      }
+    },
+    "train_folds": [
+      1,
+      2,
+      3,
+      4
+    ],
+    "val_folds": [
+      0
+    ]
+  }
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:32943c3a71cf0ff9e22cdf441e1823af4ab1faf05b5127fb0d3460a7cb6cbb8a
+oid sha256:e4af8ef6e7b13c8bb5754f00c77a422235a4d4c2031ed53abc9fa150e6997a30
 size 713881052