| { | |
| "architectures": [ | |
| "MrPongForRL" | |
| ], | |
| "model_type": "mr_pong", | |
| "obs_dim": 12, | |
| "action_dim": 3, | |
| "hidden_dims": [ | |
| 160, | |
| 160 | |
| ], | |
| "activation": "tanh", | |
| "torch_dtype": "float32", | |
| "global_step": 9975808, | |
| "auto_map": { | |
| "AutoConfig": "configuration_mr_pong.MrPongConfig", | |
| "AutoModel": "modeling_mr_pong.MrPongForRL" | |
| } | |
| } |