InternLM/doc/code-docs/source
Wenwen Qu 655e9dae40
Feat(norm)/support fused precision (#319)
* add fused precision support for norm

* refactor code

* refactor code

* change the granularity of hook

* fix bugs if self.model is ModuleList

* add dtype condition for post hook

* refactor code for split group

* refactor code for pre/post hook

* refactor code for split group

* remove fp32 hook for norm

* unit tests for fused precision

* add doc for fused precision

* add doc for En. version

* reformat docs

* Update mixed_precision.rst

* Update mixed_precision.po

* update mixed_precision.po
2023-09-26 20:39:55 +08:00
..
example docs(doc/code-docs): support zh cn readthedocs (#289) 2023-09-07 16:11:08 +08:00
checkpoint.rst feat(doc/code-docs): add checkpoint save/load usage doc (#311) 2023-09-22 18:45:33 +08:00
conf.py docs(doc/code-docs): add figure for training docs (#307) 2023-09-15 15:22:22 +08:00
index.rst Feat(norm)/support fused precision (#319) 2023-09-26 20:39:55 +08:00
initialize.rst docs(doc/code-docs): add figure for training docs (#307) 2023-09-15 15:22:22 +08:00
install.md docs(doc/code-docs): support zh cn readthedocs (#289) 2023-09-07 16:11:08 +08:00
mixed_precision.rst Feat(norm)/support fused precision (#319) 2023-09-26 20:39:55 +08:00
monitor.rst doc(monitor): add light monitoring doc (#352) 2023-09-25 19:28:09 +08:00
parallel.rst Feat(norm)/support fused precision (#319) 2023-09-26 20:39:55 +08:00
profiler.rst docs(doc/code-docs): add figure for training docs (#307) 2023-09-15 15:22:22 +08:00
qa.rst docs(doc/code-docs): add figure for training docs (#307) 2023-09-15 15:22:22 +08:00
training.rst docs(doc/code-docs): support zh cn readthedocs (#289) 2023-09-07 16:11:08 +08:00
usage.md docs(doc/code-docs): support zh cn readthedocs (#289) 2023-09-07 16:11:08 +08:00