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