Skip to content

Commit 975ed32

Browse files
1 parent 1be97cc commit 975ed32

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

docs/source/amp.rst

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@ override the defaults established by the ``opt_level``.
2727
Example::
2828

2929
# Declare model and optimizer as usual
30-
model = torch.nn.Linear(D_in, D_out).cuda().half()
30+
model = torch.nn.Linear(D_in, D_out).cuda()
3131
optimizer = torch.optim.SGD(model.parameters(), lr=1e-3)
3232

3333
# Allow Amp to perform casts as required by the opt_level

0 commit comments

Comments
 (0)