未验证 提交 c466de67 编写于 作者: W wangxinxin08 提交者: GitHub

fix bug while training in single gpu (#2140)

上级 7b0a4b30
...@@ -167,6 +167,8 @@ class Trainer(object): ...@@ -167,6 +167,8 @@ class Trainer(object):
if self._nranks > 1: if self._nranks > 1:
model = paddle.DataParallel(self.model) model = paddle.DataParallel(self.model)
else:
model = self.model
self.status.update({ self.status.update({
'epoch_id': self.start_epoch, 'epoch_id': self.start_epoch,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册