| { |
| "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 |
| ] |
| ] |
| } |
