调用模型
def save_model(model,file_name):
print('保存模型文件到：{}'.format(file_name))
torch.save(model, file_name)
print('模型文件保存完毕')
# with open(file_name,'wb') as f:
# pickle.dump(model,f)
def load_model(file_name,device):
return torch.load(file_name, map_location=device)
# with open(file_name,'rb') as f:
# return pickle.load(file_name)
Python 文章被收录于专栏
Python由荷兰数学和计算机科学研究学会的吉多·范罗苏姆于1990年代初设计，作为一门叫做ABC语言的替代品。Python提供了高效的高级数据结构，还能简单有效地面向对象编程。Python语法和动态类型，以及解释型语言的本质，使它成为多数平台上写脚本和快速开发应用的编程语言，随着版本的不断更新和语言新功能的添加，逐渐被用于独立的、大型项目的开发
阿里巴巴灵犀互娱公司福利 668人发布
查看8道真题和解析