fix span output

TNTwise · Oct 27, 2024 · 4bcca51 · 4bcca51
1 parent 287cd39
commit 4bcca51
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/backend/src/spandrel/architectures/SPAN/__arch/span.py b/backend/src/spandrel/architectures/SPAN/__arch/span.py
@@ -190,9 +190,11 @@ def update_params(self):
         self.eval_conv.bias.data = self.bias_concat.contiguous()  # type: ignore
 
     def forward(self, x):
+        self.update_params()
         out = self.eval_conv(x)
 
         if self.has_relu:
+
             out = F.leaky_relu(out, negative_slope=0.05)
         return out