学習中にexit code 137 (SIGKILL)が発生したときの対処

比較的大規模な学習器で1エポックが終わらないうちに Process exit code 137 (SIGKILL) で終了。原因は、lossをテンソルの状態で蓄積したために計算グラフ全体が蓄積されることになりメモリーが枯渇しました。