
Pytorch
文章平均质量分 59
不堪沉沦
业精于勤而荒于嬉,行成于思而毁于随。
展开
专栏收录文章
- 默认排序
- 最新发布
- 最早发布
- 最多阅读
- 最少阅读
-
torch.load 出现 AttributeError: Can‘t get attribute ‘Net‘ on module ‘__main__‘
torch.load 出现 AttributeError: Can't get attribute 'Net' on module '__main__'问题解决方案问题最近,将已经训练好的模型保存下来后,通过torch.load(model_path)方法读取时,发现没办法正常运行,抛出如下错误: AttributeError: Can't get attribute 'Net' on module '__main__'我直接好家伙,骂骂咧咧去搜为啥。报错原因: torch.load()方法所原创 2021-09-01 20:49:32 · 14809 阅读 · 10 评论 -
detach()、data、with no_grad()、requires_grad之间关系
detach()、.data、with no_grad()、requires_grad之间关系[@TOC](detach ()、.data、with no_grad()、requires_grad之间关系)原创 2020-07-18 20:48:15 · 4887 阅读 · 5 评论 -
看完秒懂torch.stack()
torch.stack ()在这里插入图片描述一、准备数据二、dim=0三、dim=1四、dim=2一、准备数据首先把基本的数据准备好:import torchimport numpy as np# 创建3*3的矩阵,a、ba=np.array([[1,2,3],[4,5,6],[7,8,9]])b=np.array([[10,20,30],[40,50,60],[70,80,90...原创 2020-03-31 22:26:48 · 4821 阅读 · 5 评论