About the performance improvement compared with Tensorflow implementation

[LiFXe]

Hi, thank you for the implementation very much, it helps me a lot! BTW, I am really confused about the performance improvement comparing the PyTorch implementation with the original Tensorflow implementation. I would be very grateful if you could give me some guidance, thank you in advance！

[yuqirose]

What is your confusion? Could you elaborate your questions?

[LiFXe]

What is your confusion? Could you elaborate your questions?
This Pytorch implementation performs much better than the original Tensorflow implementation in MAE and other metrics, I am really confused about this.

[yuqirose]

#3 (comment)