From 8f195c2ff8bf02af45abd3af983e242717f75d99 Mon Sep 17 00:00:00 2001 From: stefanradev93 Date: Wed, 12 Jun 2024 05:49:58 -0400 Subject: [PATCH] Slight semantic change --- .../experimental/networks/deep_set/equivariant_module.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/bayesflow/experimental/networks/deep_set/equivariant_module.py b/bayesflow/experimental/networks/deep_set/equivariant_module.py index f911f1bc..7fd5e3d5 100644 --- a/bayesflow/experimental/networks/deep_set/equivariant_module.py +++ b/bayesflow/experimental/networks/deep_set/equivariant_module.py @@ -69,7 +69,7 @@ def __init__( layer = layers.SpectralNormalization(layer) self.equivariant_fc.add(layer) - self.ln = layers.LayerNormalization() if layer_norm else None + self.layer_norm = layers.LayerNormalization() if layer_norm else None def call(self, input_set: Tensor, **kwargs) -> Tensor: """Performs the forward pass of a learnable equivariant transform. @@ -101,8 +101,8 @@ def call(self, input_set: Tensor, **kwargs) -> Tensor: # Pass through final equivariant transform + residual output_set = input_set + self.equivariant_fc(output_set, training=training) - if self.ln is not None: - output_set = self.ln(output_set, training=training) + if self.layer_norm is not None: + output_set = self.layer_norm(output_set, training=training) return output_set