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

removed unnessassary parts of config
Gitea Actions For AIIA / Explore-Gitea-Actions (push) Successful in 26s Details

This commit is contained in:
Falko Victor Habel 2025-06-02 18:37:29 +02:00
parent f255af1b03
commit 276f02d448
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -17,11 +17,11 @@ class aiuNN(PreTrainedModel):
# Enhanced approach
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(
in_channels=self.aiia_model.config.hidden_size,
in_channels=self.config.hidden_size,
out_channels=out_channels,
kernel_size=self.aiia_model.config.kernel_size,
kernel_size=self.config.kernel_size,
padding=1
)
self.pixel_shuffle = nn.PixelShuffle(scale_factor)