Skip to content

GitLab

Commit 7243a7d3 authored by A. Unique TensorFlower's avatar A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 470819579
parent 34562fe4
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/vision/modeling/layers/nn_layers.py
View file @ 7243a7d3
...@@ -241,7 +241,7 @@ class StochasticDepth(tf.keras.layers.Layer): ...@@ -241,7 +241,7 @@ class StochasticDepth(tf.keras.layers.Layer):
self._drop_rate = stochastic_depth_drop_rate self._drop_rate = stochastic_depth_drop_rate
def get_config(self): def get_config(self):
config = {'drop_rate': self._drop_rate} config = {'stochastic_depth_drop_rate': self._drop_rate}
base_config = super(StochasticDepth, self).get_config() base_config = super(StochasticDepth, self).get_config()
return dict(list(base_config.items()) + list(config.items())) return dict(list(base_config.items()) + list(config.items()))
......
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