Skip to content

Residual training in the downscaling_only mode + add dates to the Item class#174

Merged
LBerth merged 23 commits intometeofrance:mainfrom
Metriida:downscaling_train_on_residuals
Nov 3, 2025
Merged

Residual training in the downscaling_only mode + add dates to the Item class#174
LBerth merged 23 commits intometeofrance:mainfrom
Metriida:downscaling_train_on_residuals

Conversation

@Metriida
Copy link
Contributor

  • add dates to the Item class
  • store metrics in a list (makes adding new metrics easier)
  • the output of the model in downscaling mode is the residual between the coarse grid parameter and the fine grid parameter

@LBerth LBerth merged commit fc54b67 into meteofrance:main Nov 3, 2025
1 check passed
Comment on lines +321 to +325
self.list_metrics = [
self.acc_metric,
self.psd_plot_metric,
self.rmse_psd_plot_metric,
]
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Instead of a list, perhaps you could use a MetricCollection? This would simplify the subsequent calls to update() and compute().

See https://lightning.ai/docs/torchmetrics/stable/pages/overview.html#metriccollection

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants