Pytorch 错误：再次尝试向后遍历图形，但已释放保存的中间结果

问题描述

我有一种情况，对于每个小批量，我有多个嵌套数据，需要为其训练模型。

for idx,batch in enumerate(train_DataLoader):
data = batch.get("data").squeeze(0)
op = torch.zeros(size) #zero_initializations
for i in range(data.shape[0]):
    optimizer.zero_grad()
    current_data = data[i,...]
    start_to_current_data = data[:i+1,...]
    target =  some_transformation_func(start_to_current_data)
    op = model(current_data,op)
    loss = criterion(op,target)
    loss.backward()
    optimizer.step()

但是当我开始训练时，我收到以下错误RuntimeError：尝试再次向后浏览图表，但保存的中间结果已被释放。第一次向后调用时指定retain_graph=True。设置 retain_graph=True 会增加内存使用量，我无法训练模型。我该如何解决这个问题。

解决方法

暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！

如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@）