Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
apex
Commits
f8557569
Commit
f8557569
authored
Jun 24, 2019
by
Michael Carilli
Browse files
Docstring for multiple losses
parent
f29b3f8d
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
1 deletion
+1
-1
docs/source/advanced.rst
docs/source/advanced.rst
+1
-1
No files found.
docs/source/advanced.rst
View file @
f8557569
...
@@ -94,7 +94,7 @@ receive gradients.
...
@@ -94,7 +94,7 @@ receive gradients.
If, for a given backward pass, there'
s
only
one
optimizer
whose
params
are
about
to
receive
gradients
,
If, for a given backward pass, there'
s
only
one
optimizer
whose
params
are
about
to
receive
gradients
,
you
may
pass
that
optimizer
directly
to
``
amp
.
scale_loss
``.
Otherwise
,
you
must
pass
the
you
may
pass
that
optimizer
directly
to
``
amp
.
scale_loss
``.
Otherwise
,
you
must
pass
the
list
of
optimizers
whose
params
are
about
to
receive
gradients
::
list
of
optimizers
whose
params
are
about
to
receive
gradients
.
Example
with
3
losses
and
2
optimizers
::
#
loss0
accumulates
gradients
only
into
params
owned
by
optim0
:
#
loss0
accumulates
gradients
only
into
params
owned
by
optim0
:
with
amp
.
scale_loss
(
loss0
,
optim0
)
as
scaled_loss
:
with
amp
.
scale_loss
(
loss0
,
optim0
)
as
scaled_loss
:
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment