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, ),
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, ),