Skip to content

Commit

Permalink
corrected based on new prioritized_sampling approach
Browse files Browse the repository at this point in the history
  • Loading branch information
AOS55 committed Oct 25, 2022
1 parent 522f899 commit 9caf0f8
Showing 1 changed file with 4 additions and 7 deletions.
11 changes: 4 additions & 7 deletions configs/prioritized_sampling.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -11,15 +11,12 @@ frame_stack: 1 # only works if obs_type=pixels
action_repeat: 1 # set to 2 for pixels
discount: 0.99
skill_dim: 500
random_start: true
# train settings
num_train_frames: 2000010
num_seed_frames: 4000
# eval
eval_every_frames: 10000
num_eval_episodes: 10
# sample settings
num_sample_episodes: 1000
num_prioritize_sample_episodes: 100
num_sample_episodes: 50000
num_prioritize_sample_episodes: 1200
data_type: unsupervised # either offline_rl or unsupervised changes the data collected from the environment
# pretrained
snapshot_ts: 500000
Expand All @@ -33,7 +30,7 @@ update_encoder: false # can be either true or false depending if we want to fine
# misc
seed: 1
device: cuda
save_video: true
save_video: false
save_train_video: false
save_to_data: true
use_tb: false
Expand Down

0 comments on commit 9caf0f8

Please sign in to comment.