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

Merge pull request 'removed unnessassary parts of config' (#20) from feat/model_fix into main
Run VectorLoader Script / Explore-Gitea-Actions (push) Successful in 9s Details
Gitea Actions For AIIA / Explore-Gitea-Actions (push) Successful in 25s Details 

Reviewed-on: #20
This commit is contained in:
Falko Victor Habel 2025-06-02 16:43:58 +00:00
parent a9b35dc4e0 276f02d448
commit cb64f7e871
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)