未验证 提交 70d4a0ba 编辑于 作者: romerojosh's avatar romerojosh 提交者: GitHub
浏览文件

Fix typo in contrib FusedLamb. (#1172)

上级 76daa454
......@@ -111,7 +111,7 @@ class FusedLAMB(torch.optim.Optimizer):
continue
if p.dtype == torch.float32:
g_all_32.append(p.grad.data)
elif p.dytpe == torch.float16:
elif p.dtype == torch.float16:
g_all_16.append(p.grad.data)
else:
raise RuntimeError('FusedLAMB only support fp16 and fp32.')
......
Supports Markdown
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册