def METHOD_NAME(name):
    # ms.set_context(mode=ms.PYNATIVE_MODE)
    bs = 2
    c = 10
    model = create_model(model_name=name, num_classes=c)
    input_size = get_pretrained_cfg_value(model_name=name, cfg_key="input_size")
    if input_size:
        input_size = (bs,) + tuple(input_size)
    else:
        input_size = (bs, 3, 224, 224)
    dummy_input = Tensor(np.random.rand(*input_size), dtype=ms.float32)
    y = model(dummy_input)
    assert y.shape == (bs, 10), "output shape not match"