Merge pull request #101 from stefanradev93/Development

Add summary_dim attribute to transformers
stefanradev93 · Sep 12, 2023 · 2bd8744 · 2bd8744
2 parents 1419f3a + a350ec1
commit 2bd8744
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/bayesflow/summary_networks.py b/bayesflow/summary_networks.py
@@ -148,6 +148,7 @@ def __init__(
 
  # Final output reduces representation into a vector of length summary_dim
  self.output_layer = Dense(summary_dim)
+ self.summary_dim = summary_dim
 
  def call(self, x, **kwargs):
  """Performs the forward pass through the transformer.
@@ -269,6 +270,8 @@ def __init__(
  summary_dim, attention_settings, num_dense_fc, dense_settings, use_layer_norm, num_seeds
  )
 
+ self.summary_dim = summary_dim
+
  def call(self, x, **kwargs):
  """Performs the forward pass through the set-transformer.