Hi @ptrblck, I tested it on the newer version of PyTorch (1.4.0), but still unable to run it. I ran it on google colab this time, so I cannot see exactly what the error is, but it keeps crashing on GPU although it runs ok on CPU.