New default config
This commit is contained in:
parent
68fbf875c7
commit
b7ad796671
34
config.yaml
34
config.yaml
@ -1,19 +1,32 @@
|
||||
name: DEFAULT
|
||||
project: alpha
|
||||
|
||||
vars:
|
||||
foo: bar
|
||||
|
||||
slurm:
|
||||
name: False
|
||||
partition: "single"
|
||||
num_parallel_jobs: 64
|
||||
cpus-per-task: 1
|
||||
name: '{config[name]}'
|
||||
partitions:
|
||||
- SINGLE
|
||||
num_parallel_jobs: 50
|
||||
cpus_per_task: 1
|
||||
mem-per-cpu: 3000
|
||||
time: 1440 # in minutes
|
||||
ntasks: 1
|
||||
sh_lines: ['']
|
||||
|
||||
repetitions: 3
|
||||
agents_per_job: 3
|
||||
reps_per_agent: 1
|
||||
#runner: sb3
|
||||
runner: debug
|
||||
|
||||
total_timesteps: 10000
|
||||
scheduler:
|
||||
repetitions: 3
|
||||
agents_per_job: 3
|
||||
reps_per_agent: 1
|
||||
|
||||
wandb:
|
||||
project: '{config[name]}'
|
||||
sync_tensorboard: True
|
||||
monitor_gym: True
|
||||
save_code: False
|
||||
|
||||
video:
|
||||
enable: True
|
||||
@ -34,10 +47,11 @@ env:
|
||||
normalize_rew: True
|
||||
num_envs: 1
|
||||
env_args:
|
||||
more_obs:True
|
||||
more_obs: True
|
||||
|
||||
algo:
|
||||
name: PPO
|
||||
total_timesteps: 10000
|
||||
policy_name: MlpPolicy
|
||||
n_steps: 4096
|
||||
vf_coef: 1.0e-5
|
||||
|
Loading…
Reference in New Issue
Block a user