commit all
This commit is contained in:
@@ -44,7 +44,7 @@ class AutoEncoder(tf.keras.Model):
|
||||
self.decoder = Decoder(decoder_input, decoder_depths, decoder_dims, kernel_size=kernel_size)
|
||||
|
||||
|
||||
def call(self, inputs, training=None, use_mask=True):
|
||||
def call(self, inputs, training, use_mask=True):
|
||||
if use_mask:
|
||||
x, mask = inputs
|
||||
else:
|
||||
|
||||
Reference in New Issue
Block a user