b34ec7822fae93647fb7b82b33a7d776ab4460f4,tests/unittests/test_model_zoo.py,,test_imagenet_models,#,49

Before Change

              "se_resnet101_v1", "se_resnet152_v1",
              "se_resnet18_v2", "se_resnet34_v2", "se_resnet50_v2",
              "se_resnet101_v2", "se_resnet152_v2"]
    for model in models:
        net = gcv.model_zoo.get_model(model)
        with warnings.catch_warnings():
            warnings.simplefilter("ignore")
            net.initialize()
        net.collect_params().reset_ctx(ctx)
        net(x)
        mx.nd.waitall()
    for model in models:
        if "resnext" in model:
            net = gcv.model_zoo.get_model(model, use_se=True)
            with warnings.catch_warnings():

After Change

              "se_resnet101_v1", "se_resnet152_v1",
              "se_resnet18_v2", "se_resnet34_v2", "se_resnet50_v2",
              "se_resnet101_v2", "se_resnet152_v2"]
    _test_model_list(models, ctx, x)

@try_gpu(0)
def test_ssd_models():
    ctx = mx.context.current_context()

Instances