File size: 256 Bytes
82b2cb6
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
{
    "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
}