(btw) I tried the same execution with smaller learning rates 1e-4/1e-5/1e-6 over 150 iterations and didn’t get any errors.
Still waiting for help regarding this issue.
(my replicated post in github https://github.com/pytorch/pytorch/issues/36647 16)