chainyo
/

segformer-sidewalk

Image Segmentation

Model card Files Files and versions

chainyo commited on Jun 13, 2022

Commit

c769ff4

·

1 Parent(s): a18af9d

fix training_loop for flavors

Files changed (1) hide show

training_loop.py +2 -2

training_loop.py CHANGED Viewed

@@ -77,11 +77,11 @@ def main(
         config.num_labels = num_labels
         config.id2label = id2label
         config.label2id = {v: k for k, v in id2label_file.items()}
-        config.push_to_hub(".", repo_url=f"https://huggingface.co/ChainYo/segformer-{model_flavor}-sidewalk")
         checkpoint_path = checkpoint_callback.best_model_filepath
         model = SegformerForSemanticSegmentation.from_pretrained(checkpoint_path, config=config,)
-        model.push_to_hub(".", repo_url=f"https://huggingface.co/ChainYo/segformer-{model_flavor}-sidewalk")
 if __name__ == "__main__":

         config.num_labels = num_labels
         config.id2label = id2label
         config.label2id = {v: k for k, v in id2label_file.items()}
+        config.push_to_hub(f"flavors/b{model_flavor}", repo_url=f"https://huggingface.co/ChainYo/segformer-{model_flavor}-sidewalk")
         checkpoint_path = checkpoint_callback.best_model_filepath
         model = SegformerForSemanticSegmentation.from_pretrained(checkpoint_path, config=config,)
+        model.push_to_hub(f"flavors/b{model_flavor}", repo_url=f"https://huggingface.co/ChainYo/segformer-{model_flavor}-sidewalk")
 if __name__ == "__main__":