@macgaf  Have you tried checking the default values? For example Keras’ BatchNormalization momentum is set to default value of 0.99 (ref), while PyTorch’s is set to default value of 0.1 (ref) - you should set it to 0.01 in order to achieve close results. I would reccomend checking other parameters as well.