HansBambel / SmaAt-UNet

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

Make persistence model a proper model that can be called #28

Open HansBambel opened 6 months ago

HansBambel commented 6 months ago

The persistence model is just manually selecting the last input. It requires a separate function to be calculated. It would be better if it is a proper model that can be called just like the other models in the testing scripts.