Commits
pengwa authored and GitHub committed e365ad7f3ab
fix deadlock in model.train mode forward run only (#9960) * fix deadlock in model.train model forward run only * fix tests * clear the grad_fns before every forward run * add clean up on exit * fix * refine code comments