zjowowen commited on
Commit
0f0498e
1 Parent(s): b3bba94

Upload policy_config.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. policy_config.py +3 -1
policy_config.py CHANGED
@@ -15,7 +15,8 @@ exp_config = {
15
  'collector_env_num': 8,
16
  'evaluator_env_num': 8,
17
  'env_id': 'PongNoFrameskip-v4',
18
- 'frame_stack': 4
 
19
  },
20
  'policy': {
21
  'model': {
@@ -60,6 +61,7 @@ exp_config = {
60
  'render_freq': -1,
61
  'mode': 'train_iter'
62
  },
 
63
  'cfg_type': 'InteractionSerialEvaluatorDict',
64
  'stop_value': 20,
65
  'n_episode': 8
 
15
  'collector_env_num': 8,
16
  'evaluator_env_num': 8,
17
  'env_id': 'PongNoFrameskip-v4',
18
+ 'frame_stack': 4,
19
+ 'env_wrapper': 'atari_default'
20
  },
21
  'policy': {
22
  'model': {
 
61
  'render_freq': -1,
62
  'mode': 'train_iter'
63
  },
64
+ 'figure_path': None,
65
  'cfg_type': 'InteractionSerialEvaluatorDict',
66
  'stop_value': 20,
67
  'n_episode': 8