Pytorch複数回導出保存計算図の方法

230 ワード

backwardを呼び出すことで、自動的に導出することができます.backwardをもう一度呼び出すと、プログラムのエラーが見つかり、もう一度やることができません.これはPyTorchのデフォルトで1回の自動導出が完了すると計算図が破棄されるため,2回の自動導出には手動で1つのものを設定する必要があり,以下の例で説明する.
#             
y.backward(retain_graph=True)