Skip to content

GitLab

"docs/git@developer.sourcefind.cn:OpenDAS/vision.git" did not exist on "a9e4cea0cdb350de950b9bccd989fad19f826d8d"
Commit 8bd382fa authored by Michael Carilli's avatar Michael Carilli
Browse files

Minor docstring update

parent da515dca
Hide 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