25 lines
427 B
YAML
25 lines
427 B
YAML
general_cfg:
|
|
algo_name: A2C
|
|
device: cuda
|
|
env_name: CartPole-v1
|
|
eval_eps: 10
|
|
eval_per_episode: 5
|
|
load_checkpoint: false
|
|
load_path: tasks
|
|
max_steps: 200
|
|
mode: train
|
|
save_fig: true
|
|
seed: 1
|
|
show_fig: false
|
|
test_eps: 20
|
|
train_eps: 1000
|
|
algo_cfg:
|
|
actor_hidden_dim: 256
|
|
actor_lr: 0.0003
|
|
batch_size: 64
|
|
buffer_size: 100000
|
|
critic_hidden_dim: 256
|
|
critic_lr: 0.001
|
|
gamma: 0.99
|
|
hidden_dim: 256
|