Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/apex.git" did not exist on "7e311e4bcd9ba7d446d9c88f52143b0cd05ff228"
Commit 67ad3065 authored by Michael Carilli's avatar Michael Carilli
Browse files

Adding layernorm docs

parent 64f3d362
Hide whitespace changes
Inline Side-by-side
Showing with 20 additions and 0 deletions
docs/source/index.rst
View file @ 67ad3065
...@@ -44,6 +44,12 @@ Installation requires CUDA 9 or later, PyTorch 0.4 or later, and Python 3. Insta ...@@ -44,6 +44,12 @@ Installation requires CUDA 9 or later, PyTorch 0.4 or later, and Python 3. Insta
optimizers optimizers
.. toctree::
:maxdepth: 1
:caption: Fused Layer Norm
layernorm
.. reparameterization .. reparameterization
.. RNN .. RNN
......
docs/source/layernorm.rst 0 → 100644
View file @ 67ad3065
.. role:: hidden
:class: hidden-section
apex.normalization.fused_layer_norm
===================================
.. automodule:: apex.normalization.fused_layer_norm
.. currentmodule:: apex.normalization.fused_layer_norm
.. FusedAdam
----------
.. autoclass:: FusedLayerNorm
:members:
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment