Skip to content

GitLab

Commit 8bd382fa authored by Michael Carilli's avatar Michael Carilli
Browse files

Minor docstring update

parent da515dca
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
apex/optimizers/fused_adam.py
View file @ 8bd382fa
...@@ -3,7 +3,8 @@ import fused_adam_cuda ...@@ -3,7 +3,8 @@ import fused_adam_cuda
class FusedAdam(torch.optim.Adam): class FusedAdam(torch.optim.Adam):
"""Implements Adam algorithm. Currently GPU-only. """Implements Adam algorithm. Currently GPU-only. Requires Apex to be installed via
``python setup.py install --cuda_ext --cpp_ext``.
It has been proposed in `Adam: A Method for Stochastic Optimization`_. It has been proposed in `Adam: A Method for Stochastic Optimization`_.
......
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