This is an automated email from the ASF dual-hosted git repository.
zhasheng pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git
The following commit(s) were added to refs/heads/master by this push:
new c2a1a7c fix bug when prefetch_buffer=1 (#11793)
c2a1a7c is described below
commit c2a1a7ca3cc7fa6a83bc5878278530259289b335
Author: solin319 <[email protected]>
AuthorDate: Thu Jul 19 02:56:35 2018 +0800
fix bug when prefetch_buffer=1 (#11793)
* fix bug when prefetch_buffer=1
* Update base_module.py
---
python/mxnet/module/base_module.py | 13 +++++++------
1 file changed, 7 insertions(+), 6 deletions(-)
diff --git a/python/mxnet/module/base_module.py
b/python/mxnet/module/base_module.py
index 4b7355f..654e41b 100644
--- a/python/mxnet/module/base_module.py
+++ b/python/mxnet/module/base_module.py
@@ -514,12 +514,6 @@ class BaseModule(object):
monitor.tic()
self.forward_backward(data_batch)
self.update()
- try:
- # pre fetch next batch
- next_data_batch = next(data_iter)
- self.prepare(next_data_batch,
sparse_row_id_fn=sparse_row_id_fn)
- except StopIteration:
- end_of_batch = True
if isinstance(data_batch, list):
self.update_metric(eval_metric,
@@ -528,6 +522,13 @@ class BaseModule(object):
else:
self.update_metric(eval_metric, data_batch.label)
+ try:
+ # pre fetch next batch
+ next_data_batch = next(data_iter)
+ self.prepare(next_data_batch,
sparse_row_id_fn=sparse_row_id_fn)
+ except StopIteration:
+ end_of_batch = True
+
if monitor is not None:
monitor.toc_print()
