flax.errors.ScopeParamShapeError #10
Description
@Alescontrela
Initializer expected to generate shape (4, 86) but got shape (16, 86) instead for parameter "d_0" in "/model/BroadcastPositionBiases_0". (https://flax.readthedocs.io/en/latest/api_reference/flax.errors.html#flax.errors.ScopeParamShapeError)
this happened when running videogpt training in transformer.py at line 188. I dont know how to fix it unfortunately