Training crashes with no error when using CUDA

[Jakkes (Jakob)]

Sorry, but I am unable as I do not own a CUDA device myself. I can only run it through Google Colab.