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

fixed to load kwargs

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

2
src/aiunn/upsampler.py
View File

@ -9,7 +9,7 @@ class Upsampler(AIIA):
# Assume that base_model.config is an instance of UpsamplerConfig. # Assume that base_model.config is an instance of UpsamplerConfig.
super().__init__(base_model.config) super().__init__(base_model.config)
self.base_model = base_model self.base_model = base_model
self.config = UpsamplerConfig(self.base_model.config) self.config = UpsamplerConfig(kwargs=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) print(self.config.upsample_scale)
self.upsample = nn.Upsample( self.upsample = nn.Upsample(