pytorch上遇到的一些问题
关注
分享
扫一扫
Mariooooooooooo
这个作者很懒，什么都没留下…
展开
-
Pytorch 自定义网络参数，但网络参数数值不更新
如果我们仅仅将fw1的网络参数放到优化器中，依然采用b处开始计算梯度，fw2会有梯度值，但是由于没放到优化器中，因此fw2的网络参数是不会更新的。没有把定义的网络的参数放到优化器中因此，对应的网络部分的参数是不会更新的。原创 2023-04-17 10:22:30 · 824 阅读 · 0 评论 -
Unable to get repr for ＜class torch.Tensor＞
我遇到这个问题的原因是,数据已经放到GPU上,但是模型还在Cpu上,因此会遇到这个问题.解决方法就是,先把模型也放到GPU上,然后在把放到GPU上的数据输入到模型中.原创 2023-03-31 20:03:20 · 795 阅读 · 0 评论