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

debug print

This commit is contained in:
Falko Victor Habel 2025-02-23 20:02:31 +01:00
parent 86bb74835d
commit 74c4adbff0
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/aiunn/upsampler.py
View File

@ -11,6 +11,7 @@ class Upsampler(AIIA):
self.base_model = base_model self.base_model = base_model
self.config = UpsamplerConfig(self.base_model.config) self.config = UpsamplerConfig(self.base_model.config)
# Create the upsample layer using values from the configuration. # Create the upsample layer using values from the configuration.
print(self.config.upsample_scale)
self.upsample = nn.Upsample( self.upsample = nn.Upsample(
scale_factor=self.config.upsample_scale, scale_factor=self.config.upsample_scale,
mode=self.config.upsample_mode, mode=self.config.upsample_mode,