Help protect the Great Barrier Reef with TensorFlow on Kaggle Join Challenge

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.