Machine-Learning
/
aiuNN
1
0
Fork 0
Code Issues 2 Pull Requests Actions Packages Projects Releases Wiki Activity

debug print

This commit is contained in:
Falko Victor Habel 2025-01-31 17:14:28 +01:00
parent 325faef9d5
commit 9187ebe012
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/aiunn/finetune.py
View File

@ -274,6 +274,8 @@ class FineTuner:
# Forward pass (we'll use the model's existing architecture without adding layers) # Forward pass (we'll use the model's existing architecture without adding layers)
try: try:
features = self.model(low_ress) 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: except Exception as e:
raise RuntimeError(f"Error during forward pass: {str(e)}") raise RuntimeError(f"Error during forward pass: {str(e)}")