Skip to content

Instantly share code, notes, and snippets.

@Islandman93
Created August 7, 2017 02:02
Show Gist options
  • Save Islandman93/4c7af903af43c5c694fc986128d8028a to your computer and use it in GitHub Desktop.
Save Islandman93/4c7af903af43c5c694fc986128d8028a to your computer and use it in GitHub Desktop.
DQN nips hyperparameters
{
"agent": "DQNAgent",
"episodes": 10000,
"max_timesteps": 10000,
"preprocessing": [
{
"type": "image_resize",
"width": 84,
"height": 84
},
{
"type": "grayscale"
},
{
"type": "divide",
"scale": 255
},
{
"type": "sequence",
"length": 4
}
],
"exploration": {
"type": "epsilon_decay",
"epsilon": 1.0,
"epsilon_final": 0.1,
"epsilon_timesteps": 1e6
},
"batch_size": 32,
"memory_capacity": 100000,
"memory": {
"type": "replay",
"random_sampling": true
},
"update_frequency": 1,
"first_update": 100,
"repeat_update": 1,
"target_update_frequency": 10000,
"discount": 0.95,
"learning_rate": 0.0002,
"optimizer": {
"type": "rmsprop",
"decay": 0.99,
"epsilon": 1e-6
},
"tf_saver": false,
"tf_summary": null,
"log_level": "info",
"update_target_weight": 1.0,
"double_dqn": false,
"clip_loss": 0.0,
"network": [
{
"type": "conv2d",
"size": 16,
"window": 8,
"stride": 4,
"bias": true,
"padding": "VALID"
},
{
"type": "conv2d",
"size": 32,
"window": 4,
"stride": 2,
"bias": true,
"padding": "VALID"
},
{
"type": "flatten"
},
{
"type": "dense",
"size": 256,
"bias": true
}
]
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment