Push model using huggingface_hub.

README.md

@@ -29,7 +29,7 @@ model = smp.from_pretrained("<save-directory-or-this-repo>")
 ## Model init parameters
 ```python
 model_init_params = {
-    "encoder_name": "resnext50_32x4d",
+    "encoder_name": "resnet50",
     "encoder_depth": 5,
     "encoder_weights": "imagenet",
     "decoder_use_batchnorm": True,
@@ -46,8 +46,8 @@ model_init_params = {
 ```json
 [
     {
-        "test_per_image_iou": 0.6655173301696777,
-        "test_dataset_iou": 0.6499855518341064
+        "test_per_image_iou": 0.7155967354774475,
+        "test_dataset_iou": 0.6878355741500854
     }
 ]
 ```

config.json

@@ -1,19 +1,19 @@
-{
-  "_model_class": "Unet",
-  "activation": null,
-  "aux_params": null,
-  "classes": 1,
-  "decoder_attention_type": null,
-  "decoder_channels": [
-    256,
-    128,
-    64,
-    32,
-    16
-  ],
-  "decoder_use_batchnorm": true,
-  "encoder_depth": 5,
-  "encoder_name": "resnext50_32x4d",
-  "encoder_weights": "imagenet",
-  "in_channels": 3
+{
+  "_model_class": "Unet",
+  "activation": null,
+  "aux_params": null,
+  "classes": 1,
+  "decoder_attention_type": null,
+  "decoder_channels": [
+    256,
+    128,
+    64,
+    32,
+    16
+  ],
+  "decoder_use_batchnorm": true,
+  "encoder_depth": 5,
+  "encoder_name": "resnet50",
+  "encoder_weights": "imagenet",
+  "in_channels": 3
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:e241d294e9b7adb00a641df09c000a6c7278bd760599acd39d55f930a365e752
-size 128290964
+oid sha256:46b33928f032463f0d3cad4cbecad8bd983849f15262ef893fabd68803f76953
+size 130343012