search

HansBambel / SmaAt-UNet

PyTorch-Code for the Paper "SmaAt-UNet: Precipitation Nowcasting using a Small, Attentive UNet-Architecture"
230 stars 43 forks source link

Combine test_precip_lightning.py and calc_metrics_test_set.py #9

Open HansBambel opened 1 year ago

HansBambel commented 1 year ago

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