Push model using huggingface_hub.

README.md

@@ -19,7 +19,7 @@ parameters: "dataloader:\n  num_workers: 0\ndataset:\n  datasets:\n    test_weed
   \    student:\n      name: conv_distillator\n      num_classes: 2\n    teacher:\n\
   \      backbone: swin\n      backbone_checkpoint: checkpoints/swin_base_patch4_window12_384.pth\n\
   \      concat_support: false\n      image_size: 384\n      model_checkpoint: checkpoints/swin_fold0_pascal_modcross_soft.pt\n\
-  \      name: dcama\npush_to_hub:\n  organization: pasqualedem\n  repo_name: DistillFSS_WeedMap_DCAMA_5shot\n\
+  \      name: dcama\npush_to_hub:\n  repo_name: DistillFSS_WeedMap_DCAMA_5shot\n\
   refinement:\n  hot_parameters:\n  - model.conv1\n  - model.conv2\n  - model.conv3\n\
   \  - model.mixer1\n  - model.mixer2\n  - model.mixer3\n  - student\n  iterations_is_num_classes:\
   \ false\n  loss:\n    name: refine_distill\n  lr: 0.001\n  max_iterations: 500\n\
@@ -68,7 +68,6 @@ model:
       model_checkpoint: checkpoints/swin_fold0_pascal_modcross_soft.pt
       name: dcama
 push_to_hub:
-  organization: pasqualedem
   repo_name: DistillFSS_WeedMap_DCAMA_5shot
 refinement:
   hot_parameters:

config.json

@@ -1,3 +1,10 @@
 {
+  "dcama_args": {
+    "backbone": "swin",
+    "backbone_checkpoint": "checkpoints/swin_base_patch4_window12_384.pth",
+    "concat_support": false,
+    "image_size": 384,
+    "model_checkpoint": "checkpoints/swin_fold0_pascal_modcross_soft.pt"
+  },
   "num_classes": 2
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:9e6e6e38c496e436303f5f313e33805f681656444e12640a88d702d13a923e30
+oid sha256:b5cb447be9b6db5d43327fdec5366e3824b788302ab40b3fe1070db7d803f669
 size 554599424