search

hagabbar / vitamin_c

This will be the first official public release of the VItamin code base. VItamin is a python package for producing fast gravitational wave posterior samples.
GNU General Public License v3.0
20 stars 7 forks source link

batchnorm in q network #90

Open chrismessenger opened 3 years ago

chrismessenger commented 3 years ago

The name given to the hidden layer batchnorm is not correct. It is being labelled with the same name as used in the conversation layer batchnorm.

If this is a trainable parameter (which I'm unsure about) then this will be a bug.