From efe41b96ec5fceba2cf91505ad70905da98dc16c Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Fri, 2 Sep 2022 17:58:02 +0100 Subject: [PATCH] fixup --- aimodel/src/lib/ai/components/LayerContrastiveEncoder.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/aimodel/src/lib/ai/components/LayerContrastiveEncoder.py b/aimodel/src/lib/ai/components/LayerContrastiveEncoder.py index 524eff4..c74232e 100644 --- a/aimodel/src/lib/ai/components/LayerContrastiveEncoder.py +++ b/aimodel/src/lib/ai/components/LayerContrastiveEncoder.py @@ -45,7 +45,7 @@ class LayerContrastiveEncoder(tf.keras.layers.Layer): # self.embedding = tf.keras.layers.Dense(self.param_feature_dim) if summary_file: - summarywriter(self.encoder, append=True) + summarywriter(self.encoder, summary_file, append=True) def get_config(self): config = super(LayerContrastiveEncoder, self).get_config()