Added range computation for the torchmetrics call

mlcommons · Aug 10, 2023 · 24d175f · 24d175f
1 parent 5690df7
commit 24d175f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/GANDLF/metrics/synthesis.py b/GANDLF/metrics/synthesis.py
@@ -60,7 +60,7 @@ def peak_signal_noise_ratio(target, prediction, data_range=None, epsilon=None) -
     """
 
     if epsilon == None:
-        psnr = PeakSignalNoiseRatio(data_range=data_range)
+        psnr = PeakSignalNoiseRatio() if data_range == None else PeakSignalNoiseRatio(data_range=data_range[1]-data_range[0])
         return psnr(preds=prediction, target=target)
     else: # implementation of PSNR that does not give 'inf'/'nan' when 'mse==0'
         mse = mean_squared_error(target, prediction)