Correctly track train / validation losses#485

stefanradev93 merged 9 commits intodevbayesflow-org/bayesflow:devfrom

track-lossesbayesflow-org/bayesflow:track-lossesCopy head branch name to clipboard