Skip to content

GitLab

Unverified Commit 87aca22a authored by Max V. Irgiznov's avatar Max V. Irgiznov Committed by GitHub
Browse files

Fixed a typo (#856)

parent 5cb187f3
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
apex/optimizers/fused_lamb.py
View file @ 87aca22a
...@@ -112,7 +112,7 @@ class FusedLAMB(torch.optim.Optimizer): ...@@ -112,7 +112,7 @@ class FusedLAMB(torch.optim.Optimizer):
continue continue
if p.dtype == torch.float32: if p.dtype == torch.float32:
g_all_32.append(p.grad.data) g_all_32.append(p.grad.data)
elif p.dytpe == torch.float16: elif p.dtype == torch.float16:
g_all_16.append(p.grad.data) g_all_16.append(p.grad.data)
else: else:
raise RuntimeError('FusedLAMB only support fp16 and fp32.') raise RuntimeError('FusedLAMB only support fp16 and fp32.')
......
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