{
    "wandb_project": "walker27",
    "dataset_name": "RandomWalker2d/28dynamics-v9",
    "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": 23,
    "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,
    "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": 17,
    "action_dim": 6,
    "factor_dim": 13,
    "dataset_size": 8383200,
    "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": [
        [
            2.74,
            2.99,
            2.99,
            2.99,
            2.99,
            2.99,
            2.99,
            0.3,
            0.27,
            0.27,
            0.3,
            2.28,
            2.28
        ],
        [
            2.74,
            2.99,
            2.99,
            2.99,
            2.99,
            2.99,
            2.99,
            0.45,
            0.37,
            0.37,
            0.45,
            1.32,
            1.32
        ],
        [
            2.74,
            2.99,
            2.99,
            2.99,
            2.99,
            2.99,
            2.99,
            0.45,
            0.37,
            0.37,
            0.45,
            1.73,
            1.73
        ],
        [
            2.74,
            2.99,
            2.99,
            2.99,
            2.99,
            2.99,
            2.99,
            0.45,
            0.37,
            0.37,
            0.45,
            2.28,
            2.28
        ],
        [
            2.74,
            2.99,
            2.99,
            2.99,
            2.99,
            2.99,
            2.99,
            0.67,
            0.51,
            0.51,
            0.67,
            1.32,
            1.32
        ],
        [
            2.74,
            2.99,
            2.99,
            2.99,
            2.99,
            2.99,
            2.99,
            0.67,
            0.51,
            0.51,
            0.67,
            1.73,
            1.73
        ],
        [
            2.74,
            2.99,
            2.99,
            2.99,
            2.99,
            2.99,
            2.99,
            0.67,
            0.51,
            0.51,
            0.67,
            2.28,
            2.28
        ],
        [
            2.74,
            2.99,
            2.99,
            2.99,
            4.47,
            4.47,
            4.47,
            0.3,
            0.27,
            0.27,
            0.3,
            1.32,
            1.32
        ],
        [
            2.74,
            2.99,
            2.99,
            2.99,
            4.47,
            4.47,
            4.47,
            0.3,
            0.27,
            0.27,
            0.3,
            1.73,
            1.73
        ],
        [
            2.74,
            2.99,
            2.99,
            2.99,
            4.47,
            4.47,
            4.47,
            0.3,
            0.27,
            0.27,
            0.3,
            2.28,
            2.28
        ],
        [
            2.74,
            2.99,
            2.99,
            2.99,
            4.47,
            4.47,
            4.47,
            0.45,
            0.37,
            0.37,
            0.45,
            1.32,
            1.32
        ],
        [
            2.74,
            2.99,
            2.99,
            2.99,
            4.47,
            4.47,
            4.47,
            0.45,
            0.37,
            0.37,
            0.45,
            1.73,
            1.73
        ],
        [
            2.74,
            2.99,
            2.99,
            2.99,
            4.47,
            4.47,
            4.47,
            0.45,
            0.37,
            0.37,
            0.45,
            2.28,
            2.28
        ],
        [
            2.74,
            2.99,
            2.99,
            2.99,
            4.47,
            4.47,
            4.47,
            0.67,
            0.51,
            0.51,
            0.67,
            1.32,
            1.32
        ],
        [
            2.74,
            2.99,
            2.99,
            2.99,
            4.47,
            4.47,
            4.47,
            0.67,
            0.51,
            0.51,
            0.67,
            1.73,
            1.73
        ],
        [
            2.74,
            2.99,
            2.99,
            2.99,
            4.47,
            4.47,
            4.47,
            0.67,
            0.51,
            0.51,
            0.67,
            2.28,
            2.28
        ],
        [
            2.74,
            2.99,
            2.99,
            2.99,
            6.69,
            6.69,
            6.69,
            0.3,
            0.27,
            0.27,
            0.3,
            1.32,
            1.32
        ],
        [
            2.74,
            2.99,
            2.99,
            2.99,
            6.69,
            6.69,
            6.69,
            0.3,
            0.27,
            0.27,
            0.3,
            1.73,
            1.73
        ],
        [
            2.74,
            2.99,
            2.99,
            2.99,
            6.69,
            6.69,
            6.69,
            0.3,
            0.27,
            0.27,
            0.3,
            2.28,
            2.28
        ],
        [
            2.74,
            2.99,
            2.99,
            2.99,
            6.69,
            6.69,
            6.69,
            0.45,
            0.37,
            0.37,
            0.45,
            1.32,
            1.32
        ],
        [
            2.74,
            2.99,
            2.99,
            2.99,
            6.69,
            6.69,
            6.69,
            0.45,
            0.37,
            0.37,
            0.45,
            1.73,
            1.73
        ],
        [
            2.74,
            2.99,
            2.99,
            2.99,
            6.69,
            6.69,
            6.69,
            0.45,
            0.37,
            0.37,
            0.45,
            2.28,
            2.28
        ],
        [
            2.74,
            2.99,
            2.99,
            2.99,
            6.69,
            6.69,
            6.69,
            0.67,
            0.51,
            0.51,
            0.67,
            1.32,
            1.32
        ],
        [
            2.74,
            2.99,
            2.99,
            2.99,
            6.69,
            6.69,
            6.69,
            0.67,
            0.51,
            0.51,
            0.67,
            1.73,
            1.73
        ],
        [
            2.74,
            2.99,
            2.99,
            2.99,
            6.69,
            6.69,
            6.69,
            0.67,
            0.51,
            0.51,
            0.67,
            2.28,
            2.28
        ]
    ]
}