HansBambel/SmaAt-UNet

Combine test_precip_lightning.py and calc_metrics_test_set.py

HansBambel opened this issue · 0 comments

Currently, the loss and the other metrics are calculated using separate scripts. They can be combined.

This could be helpful: https://torchmetrics.readthedocs.io/en/stable/pages/implement.html