{
    "wandb_project": "walker27",
    "dataset_name": "RandomHopper/82dynamics-v7",
    "train_task_ids": [
        3,
        4,
        5,
        6,
        7,
        8,
        9,
        10,
        11,
        12,
        13,
        14,
        15,
        16,
        17,
        18,
        19,
        20,
        21,
        22,
        23,
        24,
        25,
        26,
        27
    ],
    "test_task_ids": [
        0,
        1,
        2
    ],
    "use_observation": false,
    "observation_function": "mask_dimensions",
    "observation_noise_std": 0.1,
    "observation_mask_dims": [
        0,
        1
    ],
    "history": 16,
    "min_visible_length": 16,
    "delta_t": 1,
    "embedding_size": 14,
    "mask_embedding": false,
    "cross_prediction": true,
    "inverse_loss_weight": 1.0,
    "forward_loss_weight": 1.0,
    "state_loss_weight": 1.0,
    "detach_embedding_for_state": true,
    "factor_loss_weight": 1.0,
    "onehot_factor": false,
    "detach_embedding_for_factor": true,
    "policy_loss_weight": 1.0,
    "detach_embedding_for_policy": true,
    "intra_traj_consistency_loss_weight": 0.0,
    "inter_traj_consistency_loss_weight": 0.0,
    "d_model": 256,
    "n_layer": 4,
    "head_hidden": 256,
    "n_head": 8,
    "d_ff": 1024,
    "dropout": 0.1,
    "adaptive_pooling_heads": 8,
    "adaptive_pooling_dropout": 0.1,
    "pos_encoding_max_len": 5000,
    "norm_z": true,
    "learning_rate": 0.0003,
    "num_epochs": 10,
    "batch_size": 128,
    "window_size": 2,
    "eval_interval": 1,
    "train_split": 0.8,
    "device": "cuda:0",
    "log_dir": "./dadp/embedding/logs/exp/",
    "seed": 42,
    "save_checkpoint_epochs": 10,
    "state_dim": 11,
    "action_dim": 3,
    "factor_dim": 8,
    "dataset_size": 4083600,
    "observation_kwargs": {
        "mask_dims": [
            0,
            1
        ]
    },
    "auto_calculated_padding": 19,
    "selected_task_ids": [
        3,
        4,
        5,
        6,
        7,
        8,
        9,
        10,
        11,
        12,
        13,
        14,
        15,
        16,
        17,
        18,
        19,
        20,
        21,
        22,
        23,
        24,
        25,
        26,
        27
    ],
    "selected_tasks": [
        [
            3.53,
            3.93,
            2.71,
            5.09,
            0.75,
            0.75,
            0.95,
            0.75
        ],
        [
            3.53,
            3.93,
            2.71,
            5.09,
            0.75,
            0.75,
            0.95,
            0.95
        ],
        [
            3.53,
            3.93,
            2.71,
            5.09,
            0.75,
            0.75,
            0.95,
            1.19
        ],
        [
            3.53,
            3.93,
            2.71,
            5.09,
            0.75,
            0.75,
            1.19,
            0.75
        ],
        [
            3.53,
            3.93,
            2.71,
            5.09,
            0.75,
            0.75,
            1.19,
            0.95
        ],
        [
            3.53,
            3.93,
            2.71,
            5.09,
            0.75,
            0.75,
            1.19,
            1.19
        ],
        [
            3.53,
            3.93,
            2.71,
            5.09,
            0.75,
            0.95,
            0.75,
            0.75
        ],
        [
            3.53,
            3.93,
            2.71,
            5.09,
            0.75,
            0.95,
            0.75,
            0.95
        ],
        [
            3.53,
            3.93,
            2.71,
            5.09,
            0.75,
            0.95,
            0.75,
            1.19
        ],
        [
            3.53,
            3.93,
            2.71,
            5.09,
            0.75,
            0.95,
            0.95,
            0.75
        ],
        [
            3.53,
            3.93,
            2.71,
            5.09,
            0.75,
            0.95,
            0.95,
            0.95
        ],
        [
            3.53,
            3.93,
            2.71,
            5.09,
            0.75,
            0.95,
            0.95,
            1.19
        ],
        [
            3.53,
            3.93,
            2.71,
            5.09,
            0.75,
            0.95,
            1.19,
            0.75
        ],
        [
            3.53,
            3.93,
            2.71,
            5.09,
            0.75,
            0.95,
            1.19,
            0.95
        ],
        [
            3.53,
            3.93,
            2.71,
            5.09,
            0.75,
            0.95,
            1.19,
            1.19
        ],
        [
            3.53,
            3.93,
            2.71,
            5.09,
            0.75,
            1.19,
            0.75,
            0.75
        ],
        [
            3.53,
            3.93,
            2.71,
            5.09,
            0.75,
            1.19,
            0.75,
            0.95
        ],
        [
            3.53,
            3.93,
            2.71,
            5.09,
            0.75,
            1.19,
            0.75,
            1.19
        ],
        [
            3.53,
            3.93,
            2.71,
            5.09,
            0.75,
            1.19,
            0.95,
            0.75
        ],
        [
            3.53,
            3.93,
            2.71,
            5.09,
            0.75,
            1.19,
            0.95,
            0.95
        ],
        [
            3.53,
            3.93,
            2.71,
            5.09,
            0.75,
            1.19,
            0.95,
            1.19
        ],
        [
            3.53,
            3.93,
            2.71,
            5.09,
            0.75,
            1.19,
            1.19,
            0.75
        ],
        [
            3.53,
            3.93,
            2.71,
            5.09,
            0.75,
            1.19,
            1.19,
            0.95
        ],
        [
            3.53,
            3.93,
            2.71,
            5.09,
            0.75,
            1.19,
            1.19,
            1.19
        ],
        [
            3.53,
            3.93,
            2.71,
            5.09,
            1.0,
            1.0,
            1.0,
            1.0
        ]
    ]
}