Skip to content

Do you have a TrainConfig for fine-tuning pi0.5 libero lora? Is this configuration correct? #974

@wanghq-thu

Description

@wanghq-thu

TrainConfig( name="pi05_libero_low_mem_finetune", model=pi0_config.Pi0Config( pi05=True, action_horizon=10, discrete_state_input=False, paligemma_variant="gemma_2b_lora", action_expert_variant="gemma_300m_lora", ), data=LeRobotLiberoDataConfig( repo_id="physical-intelligence/libero", base_config=DataConfig(prompt_from_task=True), extra_delta_transform=False, ), weight_loader=weight_loaders.CheckpointWeightLoader( "gs://openpi-assets/checkpoints/pi05_base/params" ), freeze_filter=pi0_config.Pi0Config( pi05=True, action_horizon=10, discrete_state_input=False, paligemma_variant="gemma_2b_lora", action_expert_variant="gemma_300m_lora", ).get_freeze_filter(), ema_decay=None, batch_size=8, num_train_steps=30_000, ),

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions