code: def get_kwargs(config, use_aux_batchnorm, use_aux_dropout):
      """Get parameterization for call."""
      if config.model_name == "resnet_advtrain_classification":
          apply_kwargs = {
              "use_aux_batchnorm": use_aux_batchnorm,
              "use_aux_dropout": use_aux_dropout,
          }
      elif (
          config.model_name == "vit_advtrain_multilabel_classification"
      ):
          apply_kwargs = {
              "use_aux_dropout": use_aux_dropout,
          }
      else:
          raise ValueError("Unknown model: %s" % config.model_name)
      return apply_kwargs
  
imports: from typing import Type
  from scenic.model_lib.base_models import base_model
  from scenic.projects.adversarialtraining.models import vit_advtrain
label_desc: Use image node_id 2 for calling a global function with example usage: get_kwargs(config, use_aux_batchnorm, use_aux_dropout) and returns: apply_kwargs