提交 ae2b2185 编写于 作者: L lyuwenyu 提交者: jzhang533

update

上级 87eb929f
...@@ -1703,6 +1703,7 @@ class Model(object): ...@@ -1703,6 +1703,7 @@ class Model(object):
do_eval = eval_loader is not None do_eval = eval_loader is not None
self._test_dataloader = eval_loader self._test_dataloader = eval_loader
self._accumulate = accumulate self._accumulate = accumulate
steps = self._len_data_loader(train_loader) steps = self._len_data_loader(train_loader)
...@@ -2009,7 +2010,12 @@ class Model(object): ...@@ -2009,7 +2010,12 @@ class Model(object):
model_filename=model_filename, model_filename=model_filename,
params_filename=params_filename) params_filename=params_filename)
def _run_one_epoch(self, data_loader, callbacks, mode, logs={},): def _run_one_epoch(
self,
data_loader,
callbacks,
mode,
logs={}, ):
outputs = [] outputs = []
for step, data in enumerate(data_loader): for step, data in enumerate(data_loader):
# data might come from different types of data_loader and have # data might come from different types of data_loader and have
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册