调用模型

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语法和动态类型,以及解释型语言的本质,使它成为多数平台上写脚本和快速开发应用的编程语言,随着版本的不断更新和语言新功能的添加,逐渐被用于独立的、大型项目的开发

全部评论

相关推荐

牛客54175811...:今年对双非很难。1、争取一段大厂实习经历,2、狂磕八股,3、再跑个难度提升的项目。
点赞 评论 收藏
分享
09-29 00:03
门头沟学院 Java
点赞 评论 收藏
分享
评论
13
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务