Skip to content

Commit

Permalink
Merge pull request #339 from vpratz/fix-coupling-flow-warning
Browse files Browse the repository at this point in the history
Fix unused weights warning in coupling flow
  • Loading branch information
vpratz authored Feb 26, 2025
2 parents 051f998 + 5d43376 commit 5bf9a54
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion bayesflow/networks/mlp/hidden_block.py
Original file line number Diff line number Diff line change
Expand Up @@ -51,7 +51,7 @@ def call(self, inputs: Tensor, training: bool = False, **kwargs) -> Tensor:
def build(self, input_shape):
self.dense.build(input_shape)

if input_shape[-1] != self.units:
if input_shape[-1] != self.units and self.residual:
self.projector = self.add_weight(
shape=(input_shape[-1], self.units), initializer="glorot_uniform", trainable=True, name="projector"
)
Expand Down

0 comments on commit 5bf9a54

Please sign in to comment.