File size: 546 Bytes
62e03a2
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
general_cfg:
  algo_name: BC
  device: cuda
  env_name: CartPole-v1
  eval_eps: 10
  eval_per_episode: 1
  load_checkpoint: true
  load_path: Train_CartPole-v1_BC_20230319-190431
  max_steps: 200
  mode: test
  new_step_api: true
  render: false
  save_fig: true
  seed: 1
  show_fig: false
  test_eps: 10
  train_eps: 1
  wrapper: null
algo_cfg:
  actor_hidden_dim: 256
  batch_size: 256
  continuous: false
  critic_hidden_dim: 256
  expert_path: tasks/Collect_CartPole-v1_PPO_20230319-170351/traj/traj.pkl
  lr: 0.0003
  train_iterations: 500