提交 9d7760c9 编写于 作者: Q qingqing01

update size setting in fit

上级 6e59472d
......@@ -975,7 +975,8 @@ class Model(fluid.dygraph.Layer):
metrics=self._metrics_name(), )
def _run_one_epoch(data_loader, callbacks, mode):
size = data_loader.size if hasattr(data_loader, 'size') else None
size = len(data_loader) if hasattr(data_loader,
'__len__') else None
logs = {
'steps': size,
'metrics_name': metrics_name,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册