How to change a config using create_model

[Sreyan88]

Hi there,

I wanted to know how could I change a model config using create_model. For example I need to cfg.use_abs_pos to True for MViT-v2. Thank You!

[rwightman]

@Sreyan88 I haven't settled on a way to do cleanly modify/merge or otherwise define the config externally and pass through, so best way right now is to create a new model def for you config .. ie create a modified mvitv2_small_abs_pos config

@register_model
def mvitv2_small_abs_pos(pretrained=False, **kwargs):
    return _create_mvitv2('mvitv2_small_abs_pos', pretrained=pretrained, **kwargs)

or something like

@register_model
def mvitv2_small_abs_pos(pretrained=False, **kwargs):
  return build_model_with_cfg(
          MultiScaleVit, 'mvitv2_small, pretrained,
          model_cfg=MultiScaleVitCfg(
           .... custom config
          ),
          **kwargs
   )

Open to any feedback on the use cases ... I would like to allow modifications to be merged with an existing config, or fully define configs externally via json/yaml and pass through (including via Hugging Face hub).