From 479ec9c8a6968cf264b9c993a66918a3268b8bbd Mon Sep 17 00:00:00 2001
From: rockerBOO <rockerboo@gmail.com>
Date: Tue, 15 Jul 2025 19:52:33 -0400
Subject: [PATCH] Use new wavelet-loss library
 github.com/rockerBOO/wavelet-loss

---
 train_network.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/train_network.py b/train_network.py
index 32d6640d..2eabcdf3 100644
--- a/train_network.py
+++ b/train_network.py
@@ -43,9 +43,9 @@ from library.custom_train_functions import (
     add_v_prediction_like_loss,
     apply_debiased_estimation,
     apply_masked_loss,
-    WaveletLoss
 )
 from library.utils import setup_logging, add_logging_arguments
+from wavelet_loss import WaveletLoss
 
 setup_logging()
 import logging