We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 1be97cc commit 975ed32Copy full SHA for 975ed32
docs/source/amp.rst
@@ -27,7 +27,7 @@ override the defaults established by the ``opt_level``.
27
Example::
28
29
# Declare model and optimizer as usual
30
- model = torch.nn.Linear(D_in, D_out).cuda().half()
+ model = torch.nn.Linear(D_in, D_out).cuda()
31
optimizer = torch.optim.SGD(model.parameters(), lr=1e-3)
32
33
# Allow Amp to perform casts as required by the opt_level
0 commit comments