onnxruntime/orttraining
pengwa e365ad7f3a
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
2022-01-14 13:53:29 -08:00
..
orttraining fix deadlock in model.train mode forward run only (#9960) 2022-01-14 13:53:29 -08:00
pytorch_frontend_examples
tools Enable OneHot kernel for ROCm EP and add Dockerfile for ROCm 4.3.1 (#9656) 2021-12-07 12:47:00 -08:00