def build_tf_optimizers(cfg, default_args=None):
if len(TF_OPTIMIZERS) == 0:
register_tf_optimizers()
return build_from_cfg(cfg,
registry=TF_OPTIMIZERS,
default_args=default_args)
def build_of_models(cfg, default_args=None):
return build_from_cfg(cfg,
registry=OF_MODELS,
default_args=default_args)
def build_tf_transforms(cfg, default_args=None):
return build_from_cfg(cfg,
registry=TF_TRANSFORMS,
default_args=default_args)
def build_tf_callbacks(cfg, default_args=None):
return build_from_cfg(cfg,
registry=TF_CALLBACKS,
default_args=default_args)
def build_tf_pipelines(cfg, default_args=None):
return build_from_cfg(cfg,
registry=TF_PIPELINES,
default_args=default_args)
def build_torch_models(cfg, default_args=None):
return build_from_cfg(cfg,
registry=PT_MODELS,
default_args=default_args)
def build_tokenizers(cfg, default_args=None):
return build_from_cfg(cfg, registry=TOKENIZERS, default_args=default_args)
def build_datasets(cfg, default_args=None):
return build_from_cfg(cfg, registry=DATASETS, default_args=default_args)
def build_config(cfg, default_args=None):
return build_from_cfg(cfg, registry=CONFIGS, default_args=default_args)