diff --git a/config.yaml b/config.yaml index 4b4179c..74c9fb9 100644 --- a/config.yaml +++ b/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