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

removed unnessassary parts of config #20

Merged
Fabel merged 1 commits from feat/model_fix into main 2025-06-02 16:43:59 +00:00
Conversation 0 Commits 1 Files Changed 1 +3 -3
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit 276f02d448 - Show all commits

6
src/aiunn/upsampler/aiunn.py
View File

@ -17,11 +17,11 @@ class aiuNN(PreTrainedModel):
# Enhanced approach # Enhanced approach
scale_factor = self.config.upsample_scale scale_factor = self.config.upsample_scale
out_channels = self.aiia_model.config.num_channels * (scale_factor ** 2) out_channels = self.config.num_channels * (scale_factor ** 2)
self.pixel_shuffle_conv = nn.Conv2d( self.pixel_shuffle_conv = nn.Conv2d(
in_channels=self.aiia_model.config.hidden_size, in_channels=self.config.hidden_size,
out_channels=out_channels, out_channels=out_channels,
kernel_size=self.aiia_model.config.kernel_size, kernel_size=self.config.kernel_size,
padding=1 padding=1
) )
self.pixel_shuffle = nn.PixelShuffle(scale_factor) self.pixel_shuffle = nn.PixelShuffle(scale_factor)