1.3
https://colab.research.google.com/drive/104LtQ1zIioIOMQEPgVve77m5Rd4Gm0wU 3
1.2
https://colab.research.google.com/drive/1y4LF1a90PYfKvFgQw6fCMYKtIFK07mVr 6
Everything same except torch version.
At the end nvidia-smi show 11gb and 7 gb memory used.