提交 dbaff92b authored 作者: Frederic's avatar Frederic

Added print information when we run theano tests in batch.

This print the time used for each batch.
上级 3127e645
...@@ -185,7 +185,9 @@ def run(stdout, stderr, argv, theano_nose, batch_size, time_profile, ...@@ -185,7 +185,9 @@ def run(stdout, stderr, argv, theano_nose, batch_size, time_profile,
subprocess_extra_args.update(dict( subprocess_extra_args.update(dict(
stdout=dummy_out.fileno(), stdout=dummy_out.fileno(),
stderr=dummy_out.fileno())) stderr=dummy_out.fileno()))
t0 = time.time()
subprocess.call(cmd, **subprocess_extra_args) subprocess.call(cmd, **subprocess_extra_args)
t1 = time.time()
# Recover failed test indices from the 'failed' field of the # Recover failed test indices from the 'failed' field of the
# '.noseids' file. We need to do it after each batch because # '.noseids' file. We need to do it after each batch because
# otherwise this field may get erased. We use a set because it # otherwise this field may get erased. We use a set because it
...@@ -193,8 +195,8 @@ def run(stdout, stderr, argv, theano_nose, batch_size, time_profile, ...@@ -193,8 +195,8 @@ def run(stdout, stderr, argv, theano_nose, batch_size, time_profile,
# to avoid duplicates. # to avoid duplicates.
failed = failed.union(cPickle.load(open(noseids_file, 'rb')) failed = failed.union(cPickle.load(open(noseids_file, 'rb'))
['failed']) ['failed'])
print '%s%% done (failed: %s)' % ((test_range[-1] * 100) // print '%s%% done in %.3fs (failed: %s)' % (
n_tests, len(failed)) (test_range[-1] * 100) // n_tests, t1 - t0, len(failed))
# Sort for cosmetic purpose only. # Sort for cosmetic purpose only.
failed = sorted(failed) failed = sorted(failed)
if failed: if failed:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论