Push model using huggingface_hub.

README.md

@@ -0,0 +1,9 @@
+---
+tags:
+- pytorch_model_hub_mixin
+- model_hub_mixin
+---
+
+This model has been pushed to the Hub using ****:
+- Repo: [More Information Needed]
+- Docs: [More Information Needed]

config.json

@@ -0,0 +1,47 @@
+{
+  "cfg": {
+    "arch": "MaGGIe",
+    "aspp": {
+      "in_channels": 512,
+      "out_channels": 512
+    },
+    "decoder": "res_shortcut_inst_matt_spconv_22",
+    "decoder_args": {
+      "atten_block": 2,
+      "atten_dim": 128,
+      "atten_head": 1,
+      "atten_stride": 1,
+      "detail_mask_dropout": 0.1,
+      "final_channel": 64,
+      "freeze_detail_branch": false,
+      "head_channel": 120,
+      "max_inst": 10,
+      "use_id_pe": true,
+      "warmup_detail_iter": 3000,
+      "warmup_mask_atten_iter": 0
+    },
+    "encoder": "res_shortcut_embed_29",
+    "encoder_args": {
+      "num_embed": 3,
+      "num_mask": 10,
+      "pretrained": true
+    },
+    "having_unused_params": false,
+    "loss_alpha_grad_w": 0.05,
+    "loss_alpha_lap_w": 0.05,
+    "loss_alpha_type": "l1",
+    "loss_alpha_w": 1.0,
+    "loss_atten_w": 5.0,
+    "loss_dtSSD_w": 0.0,
+    "loss_reweight_os8": true,
+    "shm": {
+      "dilation_kernel": 15,
+      "lr_scale": 0.5,
+      "max_n_pixel": 4000000,
+      "mgm_weights": ""
+    },
+    "sync_bn": true,
+    "warmup_iters": 3000,
+    "weights": "checkpoints/image/best_model.pth"
+  }
+}

model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:404802a897f11f53810313b4a0a2ab14563837d50938951970ce17d21264d7d6
+size 120146236