26 lines
436 B
YAML
26 lines
436 B
YAML
general_cfg:
|
|
algo_name: DQN
|
|
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: 10
|
|
train_eps: 100
|
|
algo_cfg:
|
|
batch_size: 64
|
|
buffer_size: 100000
|
|
epsilon_decay: 500
|
|
epsilon_end: 0.01
|
|
epsilon_start: 0.95
|
|
gamma: 0.95
|
|
hidden_dim: 256
|
|
lr: 0.0001
|
|
target_update: 800
|