Google I/O is a wrap! Catch up on TensorFlow sessions View sessions

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.