b34ec7822fae93647fb7b82b33a7d776ab4460f4,tests/unittests/test_model_zoo.py,,test_segmentation_models,#,93

Before Change

    ctx = mx.context.current_context()
    x = mx.random.uniform(shape=(2, 3, 480, 480), ctx=ctx)
    models = ["fcn_resnet50_voc", "fcn_resnet101_voc"]
    for model in models:
        net = gcv.model_zoo.get_model(model, ctx=mx.cpu())
        with warnings.catch_warnings():
            warnings.simplefilter("ignore")
            net.initialize()
        net.collect_params().reset_ctx(ctx)
        net(x)
        mx.nd.waitall()

if __name__ == "__main__":
    import nose
    nose.runmodule()

After Change

    ctx = mx.context.current_context()
    x = mx.random.uniform(shape=(2, 3, 480, 480), ctx=ctx)
    models = ["fcn_resnet50_voc", "fcn_resnet101_voc"]
    _test_model_list(models, ctx, x)

if __name__ == "__main__":
    import nose
    nose.runmodule()

Instances