cajcodes's picture
Upload config.json
82b2cb6 verified
raw
history blame contribute delete
No virus
256 Bytes
{
"model_type": "DQN",
"input_size": 100,
"hidden_sizes": [64, 64],
"output_size": 4,
"learning_rate": 0.001,
"batch_size": 64,
"gamma": 0.99,
"target_update_freq": 100,
"num_episodes": 50,
"checkpoint_freq": 10
}