RuntimeError: Unknown model (vim_tiny_patch16_224_bimambav2_final_pool_mean_abs_pos_embed_with_midclstok_div2)

[hhhhhh45]

model = create_model( 
    args.model, 
    pretrained=False,
    num_classes=args.nb_classes,
    drop_rate=args.drop,
    drop_path_rate=args.drop_path,
    drop_block_rate=None,
    img_size=args.input_size
)

thanks for your contribution. Why is it?

[MinDeng1030]

When I deleted this line: import models_mamba in main.py, it raised the same error. After I added it back, the code worked.
I guess the timm library needs to read those registered models in models_mamba.py (line 555-615).
Hope this helps.