This commit is contained in:
Starbeamrainbowlabs 2022-09-01 16:09:24 +01:00
parent cfbbe8e8cf
commit e4c95bc7e3
Signed by: sbrl
GPG key ID: 1BE5172E637709C2

View file

@ -12,7 +12,7 @@ class LayerCheeseMultipleOut(tf.keras.layers.Layer):
"""
super(LayerCheeseMultipleOut, self).__init__(**kwargs)
self.weights_temperature = tf.Variable(name="loss_temperature", shape=1, initial_value=tf.constant([0.07]))
self.weight_temperature = tf.Variable(name="loss_temperature", shape=1, initial_value=tf.constant([0.07]))
def get_config(self):
config = super(LayerCheeseMultipleOut, self).get_config()