| { | |
| "architectures": [ | |
| "TrajectoryTransformer" | |
| ], | |
| "context_length": 20, | |
| "prediction_horizon": 50, | |
| "d_model": 256, | |
| "num_heads": 8, | |
| "num_layers": 4, | |
| "dropout": 0.1, | |
| "input_dim": 6, | |
| "output_dim": 2, | |
| "coordinate_system": "frenet", | |
| "activation_function": "gelu", | |
| "max_position_embeddings": 100, | |
| "model_type": "transformer", | |
| "torch_dtype": "float32", | |
| "transformers_version": "4.36.0" | |
| } |