2019年9月29日
pytorch note
摘要： 1.模型保存与加载 1.1 #a、保存 推荐仅仅保存模型的state_dict torch.save(model.state_dict(), MODELPATH) # .pt .pth #b、加载 model = TheModelClass(*args, **kwargs) model.load_s 阅读全文
posted @ 2019-09-29 12:00 无左无右 阅读(245) 评论(0) 推荐(0) 编辑