Change default initializer for Dense (!869) · Merge requests · Vicentini Filippo / netket

Vicentini Filippo requested to merge pv/linear into master Aug 18, 2021

Created by: PhilipVinc

using lecun_normal is the default for linear layers in flax. However, lecun_normal was broken for complex numbers so i had to change the default behaviour.

We now have fixed it upstream (a few versions ago) so there is no reason to depart from flax default behaviour anymore.

Change default initializer for Dense

Merge request reports