From aff8dd43d6a7769f00a6832dec3f45ca3d0132a6 Mon Sep 17 00:00:00 2001 From: gnzng Date: Tue, 5 Nov 2024 16:54:37 -0800 Subject: [PATCH] add tv_loss to __all__ --- src/cdtools/tools/losses/losses.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/cdtools/tools/losses/losses.py b/src/cdtools/tools/losses/losses.py index bd89fce..5339b59 100644 --- a/src/cdtools/tools/losses/losses.py +++ b/src/cdtools/tools/losses/losses.py @@ -8,7 +8,7 @@ maximum likelihood metric for a system with Poisson statistics. import torch as t -__all__ = ['amplitude_mse', 'intensity_mse', 'poisson_nll'] +__all__ = ['amplitude_mse', 'intensity_mse', 'poisson_nll', 'total_variation_loss'] def amplitude_mse(intensities, sim_intensities, mask=None, normalization=None):