Discussed with Jon about results. Suggestions:
Document suspected reasons of results (Pytorch problems with CUDA)