This commit is contained in:
Starbeamrainbowlabs 2022-09-02 17:58:02 +01:00
parent f8ee0afca1
commit efe41b96ec
Signed by: sbrl
GPG key ID: 1BE5172E637709C2

View file

@ -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()