mirror of https://github.com/hpcaitech/ColossalAI
4f4d8c3656
* [fx] added apex normalization to patched modules * remove unused imports |
||
---|---|---|
.. | ||
patched_function | ||
patched_module | ||
__init__.py | ||
registry.py |