pytorch使用
pytorch代码
Rashfu
这个作者很懒,什么都没留下…
展开
-
Pytorch学习记录
加载预训练的模型 import torch # 模型文件位置 model_file = 'mobilenet.pth.tar' # 或者.pth格式的模型文件 # 创建模型对象 model = MobileNet() # 加载模型参数,若为cpu加载,则后面添加参数: map_location='cpu' ckpt = torch.load(model_file) # cpu加载方式 ckpt = torch.load(model_file, map_location='cpu')原创 2021-08-19 11:22:55 · 272 阅读 · 0 评论 -
Pytorch学习(一)------自定义C++/ATen扩展
具体的详细细节可参考此篇博客,非常的细了 作用本质 本质其实就是用C++代码写的功能封装成python可以import的功能包,可以执行的更快!! 环境要求 老老实实用python3.6,高版本编译总会有想不到的bug 手动去这个网址找适合你的版本下载安装!!! ...原创 2021-07-28 15:59:16 · 199 阅读 · 0 评论