提交 0c63f64b 编写于 作者: C cuicheng01

Update trainer.py

上级 e8c72d36
......@@ -366,7 +366,6 @@ class Trainer(object):
if self.is_rec:
out = self.model(batch[0], batch[1])
else:
self.model.eval()
out = self.model(batch[0])
# calc loss
if self.eval_loss_func is not None:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册