Hi @Ian
Did you find the cause of the issue? and how did you fix it? thanks
I am facing a similar issue. However, I get only the error:
RuntimeError: cuda runtime error (77) : an illegal memory access was encountered at /pytorch/aten/src/THC/generated/../THCTensorMathCompareT.cuh:69
when trying to access to a tensor. (such as printing it)
This issue seems to be related to the machine where it is running.