Commit 6e8da1ab authored by lucas_miranda's avatar lucas_miranda
Browse files

Replaced for loop with vectorised mapping on ClusterOverlap regularization layer

parent 2f365a48
Pipeline #101821 passed with stages
in 23 minutes and 32 seconds
......@@ -623,11 +623,12 @@ class ClusterOverlap(Layer):
tf.cast(
tf.shape(
tf.unique(
tf.squeeze(
tf.reshape(
tf.gather(
tf.cast(hard_groups, tf.dtypes.float32),
tf.where(max_groups >= self.min_confidence),
),
[-1],
),
)[0],
)[0],
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment