Attend the Women in ML Symposium on December 7 Register now

Module: tf_agents.keras_layers.permanent_variable_rate_dropout

Stay organized with collections Save and categorize content based on your preferences.

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.