Alescontrela / viper_rl

Using advances in generative modeling to learn reward functions from unlabeled videos.
MIT License
111 stars 12 forks source link

flax.errors.ScopeParamShapeError #10

Open NonsansWD opened 2 months ago

NonsansWD commented 2 months ago

@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