| { | |
| "_name_or_path": "asRobotics/ballnet", | |
| "architectures": ["BallNet"], | |
| "model_type": "ballnet", | |
| "x_dim": [6], | |
| "y_dim": [6, 2931], | |
| "h1_dim": [100, 1000], | |
| "h2_dim": [100, 1000], | |
| "torch_dtype": "float32", | |
| "layer_norm": false, | |
| "use_activation": "relu", | |
| "auto_map": { | |
| "AutoConfig": "modeling.BallNetConfig", | |
| "AutoModel": "modeling.BallNet" | |
| } | |
| } | |