graph_weather
graph_weather copied to clipboard
Fix NormalizedMSELoss and tests
Describe the bug No current tests for just NormalizedMSELoss, and a test fails because loss got deleted I guess
To Reproduce Steps to reproduce the behavior:
- Go to '...'
- Click on '....'
- Scroll down to '....'
- See error
Expected behavior A clear and concise description of what you expected to happen.
Additional context Add any other context about the problem here.
Also NormalizedMSELoss currently doesn't take out the mean of the change like in the paper, it just ignores the variances as that was giving NaNs with the current implementation.