提交 a41b201e 编写于 作者: G gaotingquan 提交者: Wei Shengyu

debug

上级 d3374e89
......@@ -94,7 +94,7 @@ class CombinedLoss(nn.Layer):
loss_dict["loss"] = paddle.add_n(list(loss_dict.values()))
if self.scaler:
self.scaler(loss_dict["loss"])
self.scaler.scale(loss_dict["loss"])
return loss_dict
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册