Skip to content

Commit

Permalink
[dev] [benchmark] DQN-Acrobot-v1/MountainCar-v0
Browse files Browse the repository at this point in the history
  • Loading branch information
johnjim0816 committed Jun 15, 2024
1 parent e04c48e commit 7bbaf36
Show file tree
Hide file tree
Showing 6 changed files with 87 additions and 143 deletions.
47 changes: 47 additions & 0 deletions presets/ClassControl/Acrobot-v1/Acrobot-v1_DQN.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,47 @@
general_cfg:
joyrl_version: 0.6.2.5
algo_name: DQN
env_name: gym
interactor_device: cpu
learner_device: cpu
mode: train
is_learner_async: false
collect_traj: false
n_interactors: 1
load_checkpoint: true
load_path: Train_Acrobot-v1_DQN_2024613
load_model_step: best
max_episode: -1
max_step: 200
seed: 1
online_eval: true
online_eval_episode: 10
model_save_fre: 500
policy_summary_fre: 100

algo_cfg:
learn_frequency: 1
value_layers:
- layer_type: linear
layer_dim: [6,256]
activation: relu
- layer_type: linear
layer_dim: [256,256]
activation: relu
- layer_type: linear
layer_dim: [256,3]
activation: none
batch_size: 64
max_buffer_size: 100000
epsilon_decay: 500
epsilon_end: 0.01
epsilon_start: 0.95
gamma: 0.99
lr: 0.0001
target_update: 4

env_cfg:
id: Acrobot-v1
render_mode: null
wrappers:
- wrapper_name: GymDiscreteActionWrapper
22 changes: 0 additions & 22 deletions presets/ClassControl/Acrobot-v1/Acrobot-v1_DQN_Test.yaml

This file was deleted.

32 changes: 0 additions & 32 deletions presets/ClassControl/Acrobot-v1/Acrobot-v1_DQN_Train.yaml

This file was deleted.

40 changes: 40 additions & 0 deletions presets/ClassControl/MountainCar-v0/MountainCar-v0_DQN.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,40 @@
general_cfg:
joyrl_version: 0.6.2.5
algo_name: DQN
env_name: gym
interactor_device: cpu
learner_device: cpu
mode: train
is_learner_async: false
collect_traj: false
n_interactors: 1
load_checkpoint: false
load_path: "Train_MountainCar-v0_DQN"
load_model_step: "best"
max_episode: -1
max_step: 200
seed: 1
online_eval: true
online_eval_episode: 10
algo_cfg:
learn_frequency: 1
value_layers:
- layer_type: linear
layer_size: [256]
activation: relu
- layer_type: linear
layer_dim: [128]
activation: none
batch_size: 64
max_buffer_size: 10000
epsilon_decay: 1500
epsilon_end: 0.01
epsilon_start: 0.98
gamma: 0.98
lr: 0.001
target_update: 10
env_cfg:
id: MountainCar-v0
render_mode: null
wrappers:
- wrapper_name: GymDiscreteActionWrapper
45 changes: 0 additions & 45 deletions presets/ClassControl/MountainCar-v0/MountainCar-v0_DQN_Test.yaml

This file was deleted.

44 changes: 0 additions & 44 deletions presets/ClassControl/MountainCar-v0/MountainCar-v0_DQN_Train.yaml

This file was deleted.

0 comments on commit 7bbaf36

Please sign in to comment.