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

Commit de8731e

Browse files
committed
Enable correctness testing for NASNet
1 parent ec7b37b commit de8731e

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

tests/applications_test.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -176,9 +176,8 @@ def test_densenet():
176176
reason='NASNets are supported only on TensorFlow')
177177
def test_nasnet():
178178
app, last_dim = random.choice(NASNET_LIST)
179-
# TODO: enable correctness testing by passing the `module` arg.
180-
# Currently it seems that weight loading fails.
181-
_test_application_basic(app, module=None)
179+
module = nasnet
180+
_test_application_basic(app, module=module)
182181
_test_application_notop(app, last_dim)
183182
_test_application_variable_input_channels(app, last_dim)
184183
_test_app_pooling(app, last_dim)

0 commit comments

Comments
 (0)