update size setting in fit

9d7760c9 · qingqing01 · 6e59472d · 9d7760c9
隐藏空白更改
内联并排

Showing with 2 addition and 1 deletion

model.py model.py +2 -1

未找到文件。
--- a/model.py
+++ b/model.py
@@ -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,