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

View File

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