I get this error as well from time to time. It’s hard to reproduce it, but it always happens in the middle of execution (so some epochs have correctly run without any trouble).
Never had this error before updating to the latest pytorch.
I’m doing one fairly complicated broadcasting operation, so I will take a look at the “Important Breakages and Workarounds” section in the docs and update.