Hi folks, I was able to run the experiments cited above by using jupyter notebook on the same instance! This is something I noticed on both my local macbook air and the aws deep learning instance, running the gpu-training using jupyter notebook is fine, running the same experiments using python .py scripts lead to runtime out of memory error! Does this hint towards any possibility of memory leaks?