diff --git a/src/aiunn/upsampler.py b/src/aiunn/upsampler.py
index efc5c6f..bc48e97 100644
--- a/src/aiunn/upsampler.py
+++ b/src/aiunn/upsampler.py
@@ -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,