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 +5
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit 73d52f733c - Show all commits

5
src/aiunn/finetune.py
View File

@ -122,12 +122,17 @@ class ModelTrainer:
# Add upscaling layer if not already present # Add upscaling layer if not already present
if not hasattr(self.model, 'upsample'): if not hasattr(self.model, 'upsample'):
# Get existing configuration values or set defaults if necessary
hidden_size = self.model.config.hidden_size hidden_size = self.model.config.hidden_size
kernel_size = self.model.config.kernel_size kernel_size = self.model.config.kernel_size
self.model.upsample = nn.Sequential( self.model.upsample = nn.Sequential(
nn.Upsample(scale_factor=2, mode='bilinear', align_corners=False), nn.Upsample(scale_factor=2, mode='bilinear', align_corners=False),
nn.Conv2d(hidden_size, 3, kernel_size=kernel_size, padding=1) nn.Conv2d(hidden_size, 3, kernel_size=kernel_size, padding=1)
) )
# Update the model's configuration with new parameters
self.model.config.upsample_hidden_size = hidden_size
self.model.config.upsample_kernel_size = kernel_size
# Initialize optimizer and loss function # Initialize optimizer and loss function
self.criterion = nn.MSELoss() self.criterion = nn.MSELoss()