google collaboratory上でpytorchでSRGANを実装したいのですが、DataLoaderのメモリが解放されていなようで、epochを回しているとメモリエラーが発生してしまいます。
batchごとにメモリを空けるためにはどのようにしたらよいか教えてくださると幸いです。
該当のコードのgithubリンクです
https://github.com/pacifinapacific/Hello-World/blob/master/Untitled0.ipynb
これは48で回して1iteration目にメモリエラーがでていますが、
バッチサイズを1/6の8にすると6iteration程度でエラーがでます。
あなたの回答
tips
プレビュー