Apppublic

jarvez/Object_representation_model-api

sourceHugging Faceupdated 6mo agoView on Hugging Face

0likes

download_weights.py20 linesDownload Raw Back to root

1from huggingface_hub import hf_hub_download2import os3 4os.makedirs("weights", exist_ok=True)5 6print("Downloading weights...")7hf_hub_download(8    repo_id="jarvez/Object_representation_model-data",9    repo_type="dataset",10    filename="results/results_100/train_100_CoMo_base_b2pi.h5",11    local_dir="."12)13 14import shutil15shutil.move(16    "results/results_100/train_100_CoMo_base_b2pi.h5",17    "weights/train_100_CoMo_base_b2pi.h5"18)19print("Weights ready.")20