This is an automated email from the ASF dual-hosted git repository.
jxie 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 6abcdbb [MXNET-171] Fix a bug that was causing training accuracy to
be printed as nan sometimes (#10437)
6abcdbb is described below
commit 6abcdbb3333979d2aa8d7c3c7253762e921ba478
Author: Indu Bharathi <indhubhara...@gmail.com>
AuthorDate: Mon Apr 9 09:56:54 2018 -0700
[MXNET-171] Fix a bug that was causing training accuracy to be printed as
nan sometimes (#10437)
* Fix a bug that was causing training accuracy to be printed as nan
sometimes.
* Avoid the additional 'arg_eval_metric' variable. There should be no
overhead except for the batch in an epoch.
* Fix lint.
* For the last batch, Capture metrics before callback and use it to print
epoch metrics
* Remove unused import
---
python/mxnet/module/base_module.py | 5 ++++-
1 file changed, 4 insertions(+), 1 deletion(-)
diff --git a/python/mxnet/module/base_module.py
b/python/mxnet/module/base_module.py
index c03f8e7..8f5fd4a 100644
--- a/python/mxnet/module/base_module.py
+++ b/python/mxnet/module/base_module.py
@@ -522,6 +522,9 @@ class BaseModule(object):
if monitor is not None:
monitor.toc_print()
+ if end_of_batch:
+ eval_name_vals = eval_metric.get_name_value()
+
if batch_end_callback is not None:
batch_end_params = BatchEndParam(epoch=epoch,
nbatch=nbatch,
eval_metric=eval_metric,
@@ -531,7 +534,7 @@ class BaseModule(object):
nbatch += 1
# one epoch of training is finished
- for name, val in eval_metric.get_name_value():
+ for name, val in eval_name_vals:
self.logger.info('Epoch[%d] Train-%s=%f', epoch, name, val)
toc = time.time()
self.logger.info('Epoch[%d] Time cost=%.3f', epoch, (toc-tic))
--
To stop receiving notification emails like this one, please contact
j...@apache.org.
