run-35
This commit is contained in:
@@ -97,6 +97,7 @@ class GarmentEmbedding(tf.keras.layers.Layer):
|
||||
|
||||
def call(self, inputs, training=None, mask=None):
|
||||
x = inputs
|
||||
x = tf.image.per_image_standardization(x)
|
||||
|
||||
x = tf.transpose(x, [0,2,3,1])
|
||||
x = self.encoder(x, training=training)
|
||||
|
||||
Reference in New Issue
Block a user