Question on number of heads in Attentive Pooler

[orrzohar]

Dear authors,
Thank you for your great work.
I wondered why you set the number of heads in the attentive pooler to be equal to that of the JEPA encoder?

jepa/evals/video_classification_frozen/eval.py

Line 184 in 13fbba8

num_heads=encoder.num_heads,

Is this a principled design decision or just a hyperparameter?
Best,
Orr