Pytorchでの逆伝播計算（loss.backward()）

Question

前提・実現したいこと

Pytorchを使って、機械学習アルゴリズムを検討しております。
学習フェーズでretain_graphに関するエラーが出力されています。

発生している問題・エラーメッセージ

RuntimeError: Trying to backward through the graph a second time, but the saved intermediate results have already been freed. Specify retain_graph=True when calling backward the first time.

該当のソースコード

#勾配の初期化
optimizer.zero_grad()
#順伝搬
output = net(data)
#損失関数の計算
loss = f.nll_loss(output,target)
train_loss += loss.item()
#逆伝播
loss.backward(retain_graph=True)

試したこと

メッセージのとおり、loss.backward(retain_graph=True)としたにもかかわらず、同じエラーが出力されます。

補足情報（FW/ツールのバージョンなど）

Pytorchのバージョンは1.7.1です。

Answer 1

よくわかりませんが、メモリを解放したら走りました。