Skip to content
This repository was archived by the owner on Nov 17, 2023. It is now read-only.

Timing output for test_factorization_module when Verbose enabled #8363

Merged
cjolivier01 merged 12 commits into from
Oct 22, 2017
Merged

Timing output for test_factorization_module when Verbose enabled #8363

Changes from all commits
Commits
Show all changes
12 commits
Select commit Hold shift + click to select a range
b770440
Timing output for test_factorization_module when Verbose enabled
Oct 20, 2017
41d14a5
Trigger build
cjolivier01 Oct 21, 2017
3d4894d
Trigger build
cjolivier01 Oct 21, 2017
cd71c44
Trigger build
cjolivier01 Oct 21, 2017
aae70bd
Misc fixes for sparse distributed training (#8345)
eric-haibin-lin Oct 21, 2017
b174451
Fix the Readme (#8369)
mbaijal Oct 21, 2017
bfb45f0
Allow test to converge (#8351)
cjolivier01 Oct 21, 2017
84ef5cf
Update cudnn_algoreg-inl.h (#7988)
solin319 Oct 21, 2017
8e661aa
[Perl] emulate Python zip() for Perl (#8192)
tlby Oct 21, 2017
73c9677
add profile option for frontend profiling to image script (#8171)
szha Oct 21, 2017
428c535
Fix Typo (classification) (#8376)
0x6a62 Oct 21, 2017
94f2f61
Merge remote-tracking branch 'apache/master' into sparse_check
cjolivier01 Oct 22, 2017
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
17 changes: 15 additions & 2 deletions tests/python/unittest/test_module.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -460,7 +460,7 @@ def test_shared_exec_group(exec_grp_shared, exec_grp_created, shared_arg_names=N
shared_arg_names=shared_arg_names, extra_args=extra_args)


def test_factorization_machine_module():
def test_factorization_machine_module(verbose=False):
""" Test factorization machine model with sparse operators """
def check_factorization_machine_module(optimizer=None, num_epochs=None):
print("check_factorization_machine_module( {} )".format(optimizer))
Expand Down Expand Up @@ -558,9 +558,22 @@ def fm(factor_size, feature_dim, init):
if num_epochs > 1:
assert(metric.get()[1] < expected_accuracy)

check_factorization_machine_module('adam')
if verbose is True:
print("============ SGD ==========================")
start = time.clock()
check_factorization_machine_module('sgd')
if verbose is True:
print("Duration: {}".format(time.clock() - start))
print("============ ADAM ==========================")
start = time.clock()
check_factorization_machine_module('adam')
if verbose is True:
print("Duration: {}".format(time.clock() - start))
print("============ ADAGRAD ==========================")
start = time.clock()
check_factorization_machine_module('adagrad')
if verbose is True:
Copy link
Copy Markdown
Member

@szha szha Oct 22, 2017

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

if verbose

print("Duration: {}".format(time.clock() - start))


def test_module_initializer():
Expand Down