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
..
test_base_model	[Fix] Fix image dtype when enable_normalize=False. (#301)	2022-06-13 21:21:19 +08:00
test_wrappers	[Feat] Support FSDP Training (#304)	2022-06-21 15:32:56 +08:00
test_averaged_model.py	[Feature] Add autocast wrapper (#307)	2022-06-22 19:49:20 +08:00
test_base_module.py	[Feature] Add ModuleList Sequential and ModuleDict (#299)	2022-06-13 13:51:07 +08:00