提交 a55f8266 编写于 作者: W WuHaobo

support for paddle1.8

上级 bcc373db
...@@ -34,7 +34,7 @@ class Loss(object): ...@@ -34,7 +34,7 @@ class Loss(object):
def _labelsmoothing(self, target): def _labelsmoothing(self, target):
if target.shape[-1] != self._class_dim: if target.shape[-1] != self._class_dim:
one_hot_target = fluid.layers.one_hot( one_hot_target = fluid.one_hot(
input=target, depth=self._class_dim) input=target, depth=self._class_dim)
else: else:
one_hot_target = target one_hot_target = target
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册