Tomas_Lysek:

method run_backward of torch._C._EngineBase


Recently, I also encountered such issues, but it seems like eventually, it is OOM issue. By reducing the batch size, it starts working as normal.