![](https://img-blog.csdnimg.cn/20201014180756724.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
Charlotte_DL
这个作者很懒,什么都没留下…
展开
-
pytorch常用函数介绍
transforms.ToTensor()transforms.ToTensor() 可以将 PIL.Image/numpy.ndarray 数据进转化为torch.FloadTensor,并归一化到[0, 1.0]:取值范围为[0, 255]的PIL.Image,转换成形状为[C, H, W],取值范围是[0, 1.0]的torch.FloadTensor;形状为[H, W, C]的numpy.ndarray,转换成形状为[C, H, W],取值范围是[0, 1.0]的torch.FloadTen原创 2022-01-04 18:31:27 · 147 阅读 · 0 评论 -
conda/pip降级pytorch版本的方法以及安装指定pytorch的方法
原文出处: https://ptorch.com/news/198.html最近pytorh更新到0.4.1版本,真的很多小伙伴反馈有代码在版本升级后不能运行;需要降级pytorch版本,所以我整理一下降级方法,方便大家以后查阅,节约各位大佬宝贵的时间!一、Conda降级pytorch版本或安装指定版本pytorch如果你是使用conda包管理,你可以很容易实现版本降级,你只需要指定版...转载 2019-01-23 09:39:46 · 30459 阅读 · 0 评论 -
pytorch保存和加载模型方法
# 保存和加载整个模型torch.save(model_object, 'model.pkl')model = torch.load('model.pkl')# 仅保存和加载模型参数(推荐使用)torch.save(model_object.state_dict(), 'params.pkl')model_object.load_state_dict(torch.load('...转载 2019-01-23 15:09:16 · 817 阅读 · 0 评论