Skip to content
This repository has been archived by the owner on Oct 31, 2023. It is now read-only.

Commit

Permalink
Merge pull request #2 from facebookresearch/fix-iss-1
Browse files Browse the repository at this point in the history
Add missing configs.
  • Loading branch information
shagunsodhani committed Mar 20, 2021
2 parents dbab5f8 + cad7073 commit eea3c99
Show file tree
Hide file tree
Showing 3 changed files with 44 additions and 5 deletions.
22 changes: 22 additions & 0 deletions config/experiment/hipbmdp.yaml
@@ -0,0 +1,22 @@
# @package _group_
name: dmcontrol
builder:
_target_: mtrl.experiment.${experiment.name}.Experiment
init_steps: 1500
num_train_steps: 1000000
eval_freq: 10000 # same as save frequency. This should be a muliple of save.buffer.size_per_chunk
num_eval_episodes: 10
should_resume: True
save:
model:
retain_last_n: 1
# setting a natural number (say 5) retains the last n (say 5) models.
# setting -1 retains all the models.
# setting 0 retains no models.
buffer:
should_save: True
size_per_chunk: 10000
num_samples_to_save: -1 # number of samples to save. Set to -1 to save all samples
save_dir: ${setup.save_dir}
save_video: False
envs_to_exclude_during_training:
22 changes: 22 additions & 0 deletions config/experiment/metaworld.yaml
@@ -0,0 +1,22 @@
# @package _group_
name: metaworld
builder:
_target_: mtrl.experiment.${experiment.name}.Experiment
init_steps: 1500
num_train_steps: 1000000
eval_freq: 10000 # same as save frequency. This should be a muliple of save.buffer.size_per_chunk
num_eval_episodes: 10
should_resume: True
save:
model:
retain_last_n: 1
# setting a natural number (say 5) retains the last n (say 5) models.
# setting -1 retains all the models.
# setting 0 retains no models.
buffer:
should_save: True
size_per_chunk: 10000
num_samples_to_save: -1 # number of samples to save. Set to -1 to save all samples
save_dir: ${setup.save_dir}
save_video: False
envs_to_exclude_during_training:
5 changes: 0 additions & 5 deletions config/logbook/mtrl.yaml
Expand Up @@ -3,8 +3,3 @@ _target_: ml_logger.logbook.make_config
write_to_console: False
logger_dir: ${setup.save_dir}
create_multiple_log_files: False
mongo_config:
host: devfair0140
port: 27017
db: project
collection: mtrl-v1

0 comments on commit eea3c99

Please sign in to comment.