{ | |
"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 | |
} | |
{ | |
"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 | |
} | |