We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 9d4c1a1 commit ea0659dCopy full SHA for ea0659d
bayesflow/networks/transformers/isab.py
@@ -107,6 +107,5 @@ def call(self, input_set: Tensor, training: bool = False, **kwargs) -> Tensor:
107
batch_size = keras.ops.shape(input_set)[0]
108
inducing_points_expanded = keras.ops.expand_dims(self.inducing_points, axis=0)
109
inducing_points_tiled = keras.ops.tile(inducing_points_expanded, [batch_size, 1, 1])
110
- print(kwargs)
111
h = self.mab0(inducing_points_tiled, input_set, training=training, **kwargs)
112
return self.mab1(input_set, h, training=training, **kwargs)
0 commit comments