Mashiro 59b0ccfe6f

[Fix] Fix pytorch version compatibility of autocast (#339) ...

* fix unit test of autocast

* fix compatiblity of unit test of optimizerwrapper

* clean code

* fix as comment

* fix docstring

2022-06-29 20:30:53 +08:00

..

test_amp.py

[Fix] Fix pytorch version compatibility of autocast (#339)

2022-06-29 20:30:53 +08:00

test_runner.py

[Feature]: Set different seed to different rank (#298)

2022-06-24 14:28:16 +08:00