请教如何保存和读取模型?
先保存在本地,然后上传到模型库中。
def save(model):
torch.save(model,'TEST_CAT_DOG/pytorch_model.pt')
YOUR_ACCESS_TOKEN = '**************************'
api = HubApi()
api.login(YOUR_ACCESS_TOKEN)
api.push_model(
model_id='wsmqdbb/TEST_CAT_DOG',
model_dir='TEST_CAT_DOG' # 本地模型目录,要求目录中必须包含configuration.json
)
下次使用的时候克隆并加载
def load():
#模型下载
!git clone https://www.modelscope.cn/*****/TEST_CAT_DOG.git
return torch.load('TEST_CAT_DOG/pytorch_model.pt')
赞0
踩0