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

MobileNetV2 #9614

Merged
merged 24 commits into from
Feb 21, 2018
Merged
Changes from 1 commit
Commits
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
Prev Previous commit
Next Next commit
use autopep8 to reformat code.
  • Loading branch information
dwSun authored and dwSun committed Feb 21, 2018
commit ea492678ee7a808d0ae059bb84320824150e882e
4 changes: 3 additions & 1 deletion tests/python/unittest/test_gluon_model_zoo.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,9 +21,11 @@
import sys
from common import setup_module, with_seed


def eprint(*args, **kwargs):
print(*args, file=sys.stderr, **kwargs)


@with_seed()
def test_models():
all_models = ['resnet18_v1', 'resnet34_v1', 'resnet50_v1', 'resnet101_v1', 'resnet152_v1',
Expand All @@ -41,7 +43,7 @@ def test_models():
test_pretrain = model_name in pretrained_to_test
model = get_model(model_name, pretrained=test_pretrain, root='model/')
data_shape = (2, 3, 224, 224) if 'inception' not in model_name else (2, 3, 299, 299)
eprint('testing forward for %s'%model_name)
eprint('testing forward for %s' % model_name)
print(model)
if not test_pretrain:
model.collect_params().initialize()
Expand Down