skill_dim

AOS55 · Oct 31, 2022 · eba2b4e · eba2b4e
1 parent c07c727
commit eba2b4e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/configs/pretrain.yaml b/configs/pretrain.yaml
@@ -9,11 +9,12 @@ domain: walker # primal task will be inferred in runtime
 obs_type: states # [states, pixels]
 frame_stack: 1 # only works if obs_type=pixels
 action_repeat: 1 # set to 2 for pixels
+
 skill_dim: 10
 discount: 0.99
 random_start: false
 # train settings
-num_train_frames: 2000010
+num_train_frames: 2000100
 num_seed_frames: 4000
 # eval
 eval_every_frames: 10000

diff --git a/configs/prioritized_sampling.yaml b/configs/prioritized_sampling.yaml
@@ -10,16 +10,16 @@ obs_type: states # [states, pixels]
 frame_stack: 1 # only works if obs_type=pixels
 action_repeat: 1 # set to 2 for pixels
 discount: 0.99
-skill_dim: 500
+skill_dim: 4
 random_start: true
 # train settings
 num_seed_frames: 4000
 # sample settings
-num_sample_episodes: 50000
-num_prioritize_sample_episodes: 1200
+num_sample_episodes: 5000
+num_prioritize_sample_episodes: 600
 data_type: unsupervised  # either offline_rl or unsupervised changes the data collected from the environment
 # pretrained
-snapshot_ts: 500000
+snapshot_ts: 1000000
 snapshot_base_dir: ./../../../data/models/
 # replay buffer
 replay_buffer_size: 1000000