From 9187ebe01289e6f346a64bb7e8e4e9c5e2b777d4 Mon Sep 17 00:00:00 2001
From: Falko Habel <falko.habel@gmx.de>
Date: Fri, 31 Jan 2025 17:14:28 +0100
Subject: [PATCH] debug print

---
 src/aiunn/finetune.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/aiunn/finetune.py b/src/aiunn/finetune.py
index 36ad40a..50acea1 100644
--- a/src/aiunn/finetune.py
+++ b/src/aiunn/finetune.py
@@ -274,6 +274,8 @@ class FineTuner:
             # Forward pass (we'll use the model's existing architecture without adding layers)
             try:
                 features = self.model(low_ress)
+                print("Features shape:", features.shape)  # Check output dimensions
+                print("High-res shape:", high_ress.shape)  # Check target dimensions
             except Exception as e:
                 raise RuntimeError(f"Error during forward pass: {str(e)}")