2023年8月7日
pytorch如何保存和加载模型
摘要： 两种方法：保存和加载参数 和 保存加载整个模型 保存和加载参数 #保存 torch.save(model.state_dict,PATH) #PATH推荐格式为.pt #加载 model=TheModelClass(*args, **kwargs ) model.load_state_dict(to 阅读全文
posted @ 2023-08-07 09:03 浪矢-CL 阅读(14) 评论(0) 推荐(0) 编辑