core.scripts.create_uma_finetune_dataset

Attributes

TEMPLATE_DIR
DATA_YAML_DIR
REGRESSION_LABEL_TO_TASK_YAML
UMA_SM_FINETUNE_YAML
parser

Functions

create_yaml(train_path, val_path, force_rms, ...)

Module Contents

core.scripts.create_uma_finetune_dataset.TEMPLATE_DIR

core.scripts.create_uma_finetune_dataset.DATA_YAML_DIR

core.scripts.create_uma_finetune_dataset.REGRESSION_LABEL_TO_TASK_YAML

core.scripts.create_uma_finetune_dataset.UMA_SM_FINETUNE_YAML

core.scripts.create_uma_finetune_dataset.create_yaml(train_path: str, val_path: str, force_rms: float, linref_coeff: list, output_dir: str, dataset_name: str, regression_tasks: str, base_model_name: str)

core.scripts.create_uma_finetune_dataset.parser