debug print
This commit is contained in:
parent
86bb74835d
commit
74c4adbff0
|
@ -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,
|
||||
|
|
Loading…
Reference in New Issue