search

ikostrikov / walk_in_the_park

MIT License
241 stars 35 forks source link

Trying to reproduce the results but failing: flax.errors.ScopeParamShapeError #6

Open mashli opened 1 year ago

mashli commented 1 year ago

I'm trying to execute the code of this project and I get the following error:

kernel = self.param('kernel', flax.errors.ScopeParamShapeError: Inconsistent shapes between value and initializer for parameter "kernel"

Does anyone know how to solve this problem？

Thank you very much.