Delete checkpoints/config_iter11.yaml
Browse files
checkpoints/config_iter11.yaml
DELETED
|
@@ -1,92 +0,0 @@
|
|
| 1 |
-
action_dim: 12
|
| 2 |
-
aux_loss_weight: 0.5
|
| 3 |
-
baselines_bc_batch_size: 256
|
| 4 |
-
baselines_bc_epochs: 50
|
| 5 |
-
baselines_bc_lr: 0.0003
|
| 6 |
-
baselines_bc_oracle_episodes_per_env: 5000
|
| 7 |
-
baselines_dqn_buffer_size: 100000
|
| 8 |
-
baselines_dt_batch_size: 256
|
| 9 |
-
baselines_dt_context_len: 64
|
| 10 |
-
baselines_dt_embed_dim: 256
|
| 11 |
-
baselines_dt_epochs: 50
|
| 12 |
-
baselines_dt_eval_max_steps: 200
|
| 13 |
-
baselines_dt_lr: 0.0003
|
| 14 |
-
baselines_dt_max_ep_len: 200
|
| 15 |
-
baselines_dt_n_heads: 4
|
| 16 |
-
baselines_dt_n_layers: 4
|
| 17 |
-
baselines_dt_oracle_episodes_per_env: 5000
|
| 18 |
-
baselines_eval_episodes_per_env: null
|
| 19 |
-
baselines_eval_freq_env_steps: 10000
|
| 20 |
-
baselines_n_envs_per_id: 2
|
| 21 |
-
baselines_output_dir: outputs/baselines
|
| 22 |
-
baselines_wandb_project: minihack-ReMDM-baselines
|
| 23 |
-
buffer_capacity: 50
|
| 24 |
-
checkpoint_dir: checkpoints
|
| 25 |
-
checkpoint_eval_episodes: 2
|
| 26 |
-
checkpoint_every_timesteps: 2500
|
| 27 |
-
collect_episodes_per_env: 5000
|
| 28 |
-
collect_num_workers: 8
|
| 29 |
-
collect_output: data/dataset.pt
|
| 30 |
-
crop_size: 9
|
| 31 |
-
curriculum_preseed: true
|
| 32 |
-
curriculum_queue_size: 100
|
| 33 |
-
dagger_batch_size: 256
|
| 34 |
-
dagger_grad_clip: 1.0
|
| 35 |
-
dagger_lr: 3.0e-05
|
| 36 |
-
device: cpu
|
| 37 |
-
diffusion_steps_collect: 5
|
| 38 |
-
diffusion_steps_eval: 10
|
| 39 |
-
dropout: 0.0
|
| 40 |
-
efficiency_multiplier: 1.5
|
| 41 |
-
ema_decay: 0.999
|
| 42 |
-
episodes_per_iteration: 2
|
| 43 |
-
eta: 0.15
|
| 44 |
-
eval_episodes_per_env: 2
|
| 45 |
-
global_gate_init: -3.0
|
| 46 |
-
grad_steps_per_iteration: 5
|
| 47 |
-
id_envs:
|
| 48 |
-
- MiniHack-Room-Random-5x5-v0
|
| 49 |
-
- MiniHack-Room-Random-15x15-v0
|
| 50 |
-
- MiniHack-Corridor-R2-v0
|
| 51 |
-
- MiniHack-MazeWalk-9x9-v0
|
| 52 |
-
id_eval_every_timesteps: 2500
|
| 53 |
-
label_smoothing: 0.0
|
| 54 |
-
loss_weight_clip: 1000.0
|
| 55 |
-
map_h: 21
|
| 56 |
-
map_w: 79
|
| 57 |
-
mask_token: 12
|
| 58 |
-
n_embd: 256
|
| 59 |
-
n_global_tokens: 8
|
| 60 |
-
n_head: 4
|
| 61 |
-
n_layer: 4
|
| 62 |
-
noise_schedule: linear
|
| 63 |
-
num_collection_workers: 0
|
| 64 |
-
num_diffusion_steps: 100
|
| 65 |
-
offline_batch_size: 256
|
| 66 |
-
offline_buffer_capacity: null
|
| 67 |
-
offline_checkpoint_every_grad_steps: null
|
| 68 |
-
offline_eval_every_grad_steps: null
|
| 69 |
-
offline_grad_clip: 1.0
|
| 70 |
-
offline_log_every: 10
|
| 71 |
-
offline_lr: 0.0003
|
| 72 |
-
offline_total_grad_steps: null
|
| 73 |
-
ood_envs:
|
| 74 |
-
- MiniHack-Room-Dark-15x15-v0
|
| 75 |
-
- MiniHack-Corridor-R5-v0
|
| 76 |
-
- MiniHack-MazeWalk-45x19-v0
|
| 77 |
-
ood_eval_every_timesteps: 2500
|
| 78 |
-
pad_token: 13
|
| 79 |
-
physics_aware_sampling: false
|
| 80 |
-
remask_strategy: conf
|
| 81 |
-
replan_every: 16
|
| 82 |
-
save_policy: true
|
| 83 |
-
seed: null
|
| 84 |
-
seq_len: 64
|
| 85 |
-
temperature: 0.5
|
| 86 |
-
top_p: 0.9
|
| 87 |
-
torch_compile: false
|
| 88 |
-
total_timesteps: 5000
|
| 89 |
-
use_amp: false
|
| 90 |
-
use_global_stream: true
|
| 91 |
-
use_wandb: false
|
| 92 |
-
weight_decay: 0.0001
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|