- 博客(3)
- 资源 (1)
- 收藏
- 关注
RSS订阅原创 pytorch报错：RuntimeError: Trying to backward through the graph a second time
简单地示意了一下，定义了两个模型model1和model2，model2需要用model1的输出作为输入。loss用于更新model2。报这个错误，网上都说因为同一个loss.backward()了两遍，但是我这里只有一个loss，所以对我没用。参考该回答（网上还有一些类似的）
2023-10-14 16:11:02
88
1
原创 训练过程中显存不断增加
如何解决pytorch程序运行时内存消耗一直增加的问题？- 浮生号的回答 - 知乎 https://www.zhihu.com/question/276797963/answer/2355051638。但是feature没有参与loss计算，导致feature没有被反向传播到，一直保留在计算图中。训练一个模型，显存一直在不断增加....肯定是什么变量一直被保存在计算图里面。经过排查发现，我从模型里返回了一个特征（因为这个特征后续要参与计算）。在将feature传出模型前，先detach()
2023-10-14 15:53:24
432
1
原创 POST http://xxxxxx net::ERR_CONNECTION_TIMED_OUT
看了一下防火墙没开，尝试开启了防火墙，就能连了，感觉好奇怪。
2023-04-03 17:26:32
1258
1
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人