ikostrikov / walk_in_the_park

MIT License
244 stars 33 forks source link

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

Open mashli opened 2 years ago

mashli commented 2 years 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.