Skip to content

Commit bef91de

Browse files
committed
Add build function to ISAB
1 parent 15e3fc5 commit bef91de

File tree

1 file changed

+3
-0
lines changed
  • bayesflow/networks/transformers

1 file changed

+3
-0
lines changed

bayesflow/networks/transformers/isab.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -107,3 +107,6 @@ def call(self, input_set: Tensor, training: bool = False, **kwargs) -> Tensor:
107107
inducing_points_tiled = keras.ops.tile(inducing_points_expanded, [batch_size, 1, 1])
108108
h = self.mab0(inducing_points_tiled, input_set, training=training, **kwargs)
109109
return self.mab1(input_set, h, training=training, **kwargs)
110+
111+
def build(self, input_shape):
112+
super().build(input_shape)

0 commit comments

Comments
 (0)