解决timm中自己@register_model注册模型时创建模型时找不到的问题

超级重点，最近自己希望利用timm中的@register_model注册自己的模型到factory中然后利用create_model创建自己注册的模型，但是出现读不到自己注册的模型，直接去创建现有仓库中有的模型，排查了半天网上没有相关的资料，最后我发现：！！！！！
一定要在创建模型的代码中把注册模型所在的文件导入，如下：
我注册的模型在A.py文件中

A.py
@register_model
def vit_mymodel(pretrained=False, **kwargs):
    model = VisionTransformer(**kwargs)
    model.default_cfg = _cfg()
    return model

那么我在创建这个模型的时候，一定要导入注册模型所在文件

from A.py import xxx
model = create_model(vit_mymodel,...)