Commit f1f98af4 authored by lucas_miranda's avatar lucas_miranda
Browse files

Implemented Montecarlo Dropout custom layer

parent 88900d60
......@@ -57,7 +57,7 @@ def compute_mmd(tensors):
# Custom layers for efficiency/losses
class MCDropout(tf.keras.layers.Dropout):
def call(self, inputs):
def call(self, inputs, **kwargs):
return super().call(inputs, training=True)
class DenseTranspose(Layer):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment