ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

Module: tf_agents.keras_layers.permanent_variable_rate_dropout

A keras layer that applies dropout both in training and serving.

Add the possibility to apply a variable dropout rate, that is, the rate parameter can be a callable.


class PermanentVariableRateDropout: Applies dropout both in training and serving, with variable dropout rate.