Upload stage2/config.json with huggingface_hub

stage2/config.json

@@ -1,12 +1,15 @@
 {
   "vocab_size": 23,
-  "hidden_dim": 128,
+  "hidden_dim": 192,
   "num_layers": 4,
-  "num_heads": 4,
+  "num_heads": 6,
   "dropout": 0.1,
   "max_seq_len": 512,
-  "mask_prob": 0.15,
+  "mask_prob": 0.22,
   "spatial_mlp_dim": 64,
+  "pooling_type": "cls",
+  "spatial_injection": "additive",
+  "position_embedding": "learnable",
   "context_dim": 16,
   "deep_sets_hidden": 32,
   "player_feature_dim": 4,
@@ -14,5 +17,5 @@
   "_expanded_vocab_size": 25,
   "_mask_token_id": 23,
   "_pad_token_id": 24,
-  "_output_dim": 144
+  "_output_dim": 208
 }