Awesome, I’m glad it’s working now!



 kneazle:

Keras model had kernel initializers on conv2d and linear layers, hence it didn’t have such a problem.


Just in case I made confusing statements: PyTorch modules also have default initializers, which are different to the ones Keras uses by default.