config.json

{
  "architectures": [
    "DribbleEncoderModelForPrediction"
  ],
  "bias": true,
  "block_size": 32,
  "dropout": 0.2,
  "feature_embedding_dims": [
    16,
    16,
    16,
    16,
    16,
    16,
    16,
    16,
    16,
    16,
    16,
    16,
    16,
    16,
    16,
    16,
    16,
    16,
    16,
    16,
    16,
    16,
    16,
    16,
    16,
    16,
    16,
    16,
    16,
    16,
    16,
    16,
    16,
    16,
    16,
    16,
    16,
    16,
    16,
    16
  ],
  "feature_vocab_sizes": [
    368,
    45,
    160,
    7,
    1179,
    14,
    130,
    122,
    61,
    80,
    500,
    32,
    62,
    377,
    55,
    69,
    419,
    40,
    69,
    49,
    35,
    23,
    42,
    46,
    54,
    79,
    493,
    31,
    65,
    379,
    53,
    67,
    416,
    36,
    65,
    42,
    28,
    23,
    49,
    48
  ],
  "model_type": "dribble_encoder",
  "n_embd": 640,
  "n_head": 16,
  "n_layer": 12,
  "torch_dtype": "float32",
  "transformers_version": "4.38.2"
}