SamSweere / xmm-superres-denoise

Deep Learning-Based Super-Resolution and De-Noising for XMM-Newton Images.
7 stars 1 forks source link

Use `CompositionalMetric` where possible #12

Closed bojobo closed 1 year ago

bojobo commented 1 year ago

Instead of initializing our own Metric implementation (metrics._Metric) with a given scaling and correction, initialise them without those values and just do scaling * _Metric + correction when appending the metric to the list of metrics used for the calculation of the loss. For now, the affected metrics are PoissonNLLLoss and VGGLoss.

The overall goal is to move away from creating our own metrics to having them implemented in torchmetrics but this will take some time.