出现报错原因:多个模型都会调用另外一个模型B的输出作为输入,然后用
的损失函数计算出梯度后更新B(A与B不是同一个优化器)出现这种报错。
当试过修改 +=类似的,和修改变量名后都失败了,最后降低pytorch版本到1.3成功解决。
出现报错原因:多个模型都会调用另外一个模型B的输出作为输入,然后用
的损失函数计算出梯度后更新B(A与B不是同一个优化器)出现这种报错。
当试过修改 +=类似的,和修改变量名后都失败了,最后降低pytorch版本到1.3成功解决。
您可能感兴趣的与本文相关的镜像
PyTorch 2.5
PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理
3074
2387
1957
7万+
3093
1817
6377