MAE Training

[ehsanngh]

Hello, and thank you for making the code publicly available.

I'm trying to train the models and ran into an issue when training the MAE. After ~160 epochs, with the validation loss of 9.83987683e-06 and the learning rate reduced to 5.63846e-06, the phase plot still shows no correlation, even though the frequency and EF plots do show some correlation (see below).

Do you have any insight into why this might be happening or suggestions on how to move forward?

Thanks for your time.