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

develop #4

Merged
Fabel merged 103 commits from develop into main 2025-03-01 21:47:17 +00:00
Conversation 0 Commits 103 Files Changed 13 +2
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit 9187ebe012 - Show all commits

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)}")