Commit 95c7dba5 authored by lucas_miranda's avatar lucas_miranda
Browse files

Changed prior initialization

parent 005e1384
Pipeline #102717 passed with stages
in 22 minutes and 8 seconds
...@@ -495,12 +495,12 @@ class ClusterOverlap(Layer): ...@@ -495,12 +495,12 @@ class ClusterOverlap(Layer):
config.update({"samples": self.samples}) config.update({"samples": self.samples})
return config return config
def call(self, inputs, training=None, **kwargs): def call(self, inputs, **kwargs):
"""Updates Layer's call method""" """Updates Layer's call method"""
encodings, categorical = inputs[0], inputs[1] encodings, categorical = inputs[0], inputs[1]
if training: if tf.test.is_gpu_available():
hard_groups = tf.math.argmax(categorical, axis=1) hard_groups = tf.math.argmax(categorical, axis=1)
max_groups = tf.reduce_max(categorical, axis=1) max_groups = tf.reduce_max(categorical, axis=1)
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