File size: 4,868 Bytes
a58beb2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 |
hydra:
run:
dir: output/training/${rl_mode}/${env_id}/${algorithm.name}/${seed}
sweep:
dir: output/training/${rl_mode}/${env_id}/${algorithm.name}/${seed}
subdir: ''
launcher:
_target_: hydra._internal.core_plugins.basic_launcher.BasicLauncher
sweeper:
_target_: hydra._internal.core_plugins.basic_sweeper.BasicSweeper
max_batch_size: null
params: null
help:
app_name: ${hydra.job.name}
header: '${hydra.help.app_name} is powered by Hydra.
'
footer: 'Powered by Hydra (https://hydra.cc)
Use --hydra-help to view Hydra specific help
'
template: '${hydra.help.header}
== Configuration groups ==
Compose your configuration from those groups (group=option)
$APP_CONFIG_GROUPS
== Config ==
Override anything in the config (foo.bar=value)
$CONFIG
${hydra.help.footer}
'
hydra_help:
template: 'Hydra (${hydra.runtime.version})
See https://hydra.cc for more info.
== Flags ==
$FLAGS_HELP
== Configuration groups ==
Compose your configuration from those groups (For example, append hydra/job_logging=disabled
to command line)
$HYDRA_CONFIG_GROUPS
Use ''--cfg hydra'' to Show the Hydra config.
'
hydra_help: ???
hydra_logging:
version: 1
formatters:
simple:
format: '[%(asctime)s][HYDRA] %(message)s'
handlers:
console:
class: logging.StreamHandler
formatter: simple
stream: ext://sys.stdout
root:
level: INFO
handlers:
- console
loggers:
logging_example:
level: DEBUG
disable_existing_loggers: false
job_logging:
version: 1
formatters:
simple:
format: '[%(asctime)s][%(name)s][%(levelname)s] - %(message)s'
handlers:
console:
class: logging.StreamHandler
formatter: simple
stream: ext://sys.stdout
file:
class: logging.FileHandler
formatter: simple
filename: ${hydra.runtime.output_dir}/${hydra.job.name}.log
root:
level: INFO
handlers:
- console
- file
disable_existing_loggers: false
env: {}
mode: MULTIRUN
searchpath: []
callbacks: {}
output_subdir: .
overrides:
hydra:
- hydra.mode=MULTIRUN
task:
- algorithm=sb3_sac
- env_id=RBC2D-medium-v0
- rl_mode=sarl
- seed=2
job:
name: test_sb3
chdir: true
override_dirname: algorithm=sb3_sac,env_id=RBC2D-medium-v0,rl_mode=sarl,seed=2
id: ???
num: ???
config_name: test_sb3
env_set: {}
env_copy: []
config:
override_dirname:
kv_sep: '='
item_sep: ','
exclude_keys: []
runtime:
version: 1.3.2
version_base: '1.3'
cwd: /cephfs/users/becktepe/git_projects/FluidGym
config_sources:
- path: hydra.conf
schema: pkg
provider: hydra
- path: /cephfs/users/becktepe/git_projects/FluidGym/runscripts/configs
schema: file
provider: main
- path: ''
schema: structured
provider: schema
output_dir: ???
choices:
cluster: local
algorithm: sb3_sac
hydra/env: default
hydra/callbacks: null
hydra/job_logging: default
hydra/hydra_logging: default
hydra/hydra_help: default
hydra/help: default
hydra/sweeper: basic
hydra/launcher: basic
hydra/output: default
verbose: false
env_id: RBC2D-medium-v0
env_kwargs: {}
eval_env_kwargs: {}
seed: 2
rl_mode: sarl
total_timesteps: 50000.0
n_eval_steps: 10
eval_freq: ${eval:${total_timesteps} // ${n_eval_steps}}
n_eval_episodes: 1
continue_training: false
checkpoint_latest: true
rl_device: cpu
eval_callback:
_target_: fluidgym.integration.sb3.EvalCallback
eval_freq: ${eval_freq}
n_eval_episodes: ${n_eval_episodes}
use_wandb: ${wandb.enable}
checkpoint_latest: ${checkpoint_latest}
wandb:
enable: true
entity: safe-autonomous-systems
project: jbecktepe-fluidgym
algorithm:
name: SAC
obj:
_target_: stable_baselines3.sac.sac.SAC
policy: MlpPolicy
learning_rate: 0.0003
buffer_size: 1000000
learning_starts: 100
batch_size: 256
tau: 0.005
gamma: 0.99
train_freq: 1
gradient_steps: -1
action_noise: null
replay_buffer_class: null
replay_buffer_kwargs: null
optimize_memory_usage: false
ent_coef: auto
target_update_interval: 1
target_entropy: auto
use_sde: false
sde_sample_freq: -1
use_sde_at_warmup: false
stats_window_size: 100
tensorboard_log: null
policy_kwargs: {}
verbose: 0
seed: ${seed}
device: cuda
_init_setup_model: true
cluster: {}
test_env_id: ${env_id}
test_env_kwargs: ${env_kwargs}
test_rl_mode: ${rl_mode}
n_test_episodes: 10
deterministic: true
save_frames: true
render_3d: true
|