guocheng66 commited on
Commit
2c287f9
1 Parent(s): 5ee6125

Upload config.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. config.json +5 -5
config.json CHANGED
@@ -17,7 +17,7 @@
17
  "max_policy_lag": 1000,
18
  "num_workers": 8,
19
  "num_envs_per_worker": 4,
20
- "batch_size": 256,
21
  "num_batches_per_epoch": 1,
22
  "num_epochs": 1,
23
  "rollout": 32,
@@ -63,7 +63,7 @@
63
  "summaries_use_frameskip": true,
64
  "heartbeat_interval": 20,
65
  "heartbeat_reporting_interval": 600,
66
- "train_for_env_steps": 4000000,
67
  "train_for_seconds": 10000000000,
68
  "save_every_sec": 120,
69
  "keep_checkpoints": 2,
@@ -128,15 +128,15 @@
128
  "wide_aspect_ratio": false,
129
  "eval_env_frameskip": 1,
130
  "fps": 35,
131
- "command_line": "--env=doom_health_gathering_supreme --num_workers=8 --num_envs_per_worker=4 --train_for_env_steps=4000000 --device=gpu --restart_behavior=overwrite --batch_size=256 --rnn_type=lstm --nonlinearity=relu --encoder_conv_architecture=resnet_impala",
132
  "cli_args": {
133
  "env": "doom_health_gathering_supreme",
134
  "restart_behavior": "overwrite",
135
  "device": "gpu",
136
  "num_workers": 8,
137
  "num_envs_per_worker": 4,
138
- "batch_size": 256,
139
- "train_for_env_steps": 4000000,
140
  "encoder_conv_architecture": "resnet_impala",
141
  "rnn_type": "lstm",
142
  "nonlinearity": "relu"
 
17
  "max_policy_lag": 1000,
18
  "num_workers": 8,
19
  "num_envs_per_worker": 4,
20
+ "batch_size": 512,
21
  "num_batches_per_epoch": 1,
22
  "num_epochs": 1,
23
  "rollout": 32,
 
63
  "summaries_use_frameskip": true,
64
  "heartbeat_interval": 20,
65
  "heartbeat_reporting_interval": 600,
66
+ "train_for_env_steps": 12000000,
67
  "train_for_seconds": 10000000000,
68
  "save_every_sec": 120,
69
  "keep_checkpoints": 2,
 
128
  "wide_aspect_ratio": false,
129
  "eval_env_frameskip": 1,
130
  "fps": 35,
131
+ "command_line": "--env=doom_health_gathering_supreme --num_workers=8 --num_envs_per_worker=4 --train_for_env_steps=12000000 --device=gpu --restart_behavior=overwrite --batch_size=512 --rnn_type=lstm --nonlinearity=relu --encoder_conv_architecture=resnet_impala",
132
  "cli_args": {
133
  "env": "doom_health_gathering_supreme",
134
  "restart_behavior": "overwrite",
135
  "device": "gpu",
136
  "num_workers": 8,
137
  "num_envs_per_worker": 4,
138
+ "batch_size": 512,
139
+ "train_for_env_steps": 12000000,
140
  "encoder_conv_architecture": "resnet_impala",
141
  "rnn_type": "lstm",
142
  "nonlinearity": "relu"