Skip to content
GitLab
菜单
项目
群组
代码片段
帮助
帮助
支持
社区论坛
快捷键
?
提交反馈
登录/注册
切换导航
菜单
打开侧边栏
wanggh
apex
提交
70d4a0ba
未验证
提交
70d4a0ba
编辑于
9月 24, 2021
作者:
romerojosh
提交者:
GitHub
9月 24, 2021
浏览文件
Fix typo in contrib FusedLamb. (#1172)
上级
76daa454
变更
1
Hide whitespace changes
Inline
Side-by-side
apex/contrib/optimizers/fused_lamb.py
浏览文件 @
70d4a0ba
...
...
@@ -111,7 +111,7 @@ class FusedLAMB(torch.optim.Optimizer):
continue
if
p
.
dtype
==
torch
.
float32
:
g_all_32
.
append
(
p
.
grad
.
data
)
elif
p
.
d
y
tpe
==
torch
.
float16
:
elif
p
.
dt
y
pe
==
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.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录