Moved calculation of mse to common_step and added logging of mean rmse#5

mfroelund wants to merge 7 commits intodmidk:dev/gefiondmidk/neural-lam:dev/gefionfrom

mfroelund:feature/log-rmse-during-train-and-valmfroelund/neural-lam:feature/log-rmse-during-train-and-valCopy head branch name to clipboard