!!python/object:openpi.training.config.TrainConfig
name: precise_packing_FunctionalFM
project_name: openpi
exp_name: pp__ee_ffm
model: !!python/object:openpi.models.pi0_config.Pi0Config
    action_dim: 32
    action_horizon: 50
    max_token_len: 48
    dtype: bfloat16
    paligemma_variant: gemma_2b
    action_expert_variant: gemma_300m
    img_variant: So400m/14
    action_loss_horizon: null
    pi05: false
    discrete_state_input: false
    use_ffm: true
    ffm_length_scale: 1.0
weight_loader: !!python/object:openpi.training.weight_loaders.CheckpointWeightLoader
    local_dir: null
    repo_id: robotgeneralist/openpi_checkpoint_mirrors
    revision: main
    checkpoint_path: pi0_base
pytorch_weight_path: null
pytorch_training_precision: bfloat16
lr_schedule: !!python/object:openpi.training.optimizer.CosineDecaySchedule
    warmup_steps: 1000
    peak_lr: 2.5e-05
    decay_steps: 30000
    decay_lr: 2.5e-06
optimizer: !!python/object:openpi.training.optimizer.AdamW
    b1: 0.9
    b2: 0.95
    eps: 1.0e-08
    weight_decay: 1.0e-10
    clip_gradient_norm: 1.0
ema_decay: 0.99
freeze_filter: !!python/name:flax.nnx.filterlib.Nothing ''
data: !!python/object:openpi.training.config.LeRobotNomagicURXDataConfig
    repo_ids: !!python/tuple
    - robotgeneralist/pp__1
    - robotgeneralist/pp__1__adversarial
    - robotgeneralist/pp__2
    - robotgeneralist/pp__2__rotations
    - robotgeneralist/pp__3__with_ft
    - robotgeneralist/pp__3__with_ft__rotations
    - robotgeneralist/pp__3__with_ft__adversarial_start
    - robotgeneralist/pp__4__with_ft
    - robotgeneralist/pp__5__with_ft__adversarial_target
    - robotgeneralist/pp__6__with_ft
    - robotgeneralist/precise-packing-2025-10-22-part-1
    - robotgeneralist/precise-packing-2025-10-22-part-2
    - robotgeneralist/precise-packing-2025-10-22-part-3
    - robotgeneralist/precise-packing-2025-10-22-part-4
    assets: !!python/object:openpi.training.config.AssetsConfig
        assets_dir: null
        asset_id: ur10e
    norm_stats_sample_ratio: 0.05
    base_config: !!python/object:openpi.training.config.DataConfig
        repo_ids: null
        asset_id: null
        norm_stats_data: null
        norm_stats_sample_ratio: 1.0
        norm_stats_dir: null
        tolerance_s: 0.0001
        per_action_index_norm_stats: false
        repack_transforms: !!python/object:openpi.transforms.Group
            inputs: !!python/tuple []
            outputs: !!python/tuple []
        data_transforms: !!python/object:openpi.transforms.Group
            inputs: !!python/tuple []
            outputs: !!python/tuple []
        model_transforms: !!python/object:openpi.transforms.Group
            inputs: !!python/tuple []
            outputs: !!python/tuple []
        use_quantile_norm: false
        action_sequence_keys: !!python/tuple
        - actions
        fourier_config: null
        prompt_from_task: true
        rlds_data_dir: null
        action_space: null
        filter_dict_path: null
    default_prompt: null
    use_poses: true
    end_effector_frame: true
    per_action_index_norm_stats: true
    fourier_config: null
    tolerance_s: 0.0001
assets_base_dir: ./assets
checkpoint_base_dir: /storage_nvme_1/robotgeneralist/checkpoints
seed: 42
batch_size: 32
num_workers: 2
num_train_steps: 25001
log_interval: 100
save_interval: 1000
keep_period: 5000
overwrite: false
resume: true
wandb_enabled: true
policy_metadata: null
fsdp_devices: 1
test_split_ratio: 0.01
test_eval_interval: 1000
test_eval_use_train_mode: false