2 Commits (486d8cda563e5054c20ecf16d5182af490805618)

Author	SHA1	Message	Date
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
Mashiro	312f264ecd	[Feature] Add autocast wrapper (#307) ... * add autocast wrapper * fix docstring * fix docstring * fix compare version * fix unit test * fix incompatible arguments * fix as comment * fix unit test * rename auto_cast to autocast	2022-06-22 19:49:20 +08:00