mirrors/mmengine
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmengine.git synced 2025-06-03 21:54:44 +08:00
Code Issues Projects Releases Wiki Activity
317 Commits 31 Branches 27 Tags
Commit Graph

2 Commits

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