From b74081975756bc9294abf45923986a13c1524f5b Mon Sep 17 00:00:00 2001 From: Falko Habel Date: Sun, 23 Feb 2025 19:27:34 +0100 Subject: [PATCH] fixed paths --- src/aiunn/finetune.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/aiunn/finetune.py b/src/aiunn/finetune.py index 3f7b941..d785030 100644 --- a/src/aiunn/finetune.py +++ b/src/aiunn/finetune.py @@ -12,7 +12,7 @@ from torchvision import transforms from tqdm import tqdm from aiia import AIIABase -from aiunn.upsampler import Upsampler +from upsampler import Upsampler # Define a simple EarlyStopping class to monitor the epoch loss. class EarlyStopping: