Keras vs pytorch

Probably not the right forum but I was trying to convert a keras to pytorch lightning. Here is the issue that I am facing

Keras vs Pytorch_lightning execution time · Discussion #8647 · PyTorchLightning/pytorch-lightning · GitHub any idea?