Internal change.

PiperOrigin-RevId: 424691273
tensorflow · Jan 27, 2022 · a033df7 · a033df7
1 parent c8a402f
commit a033df7
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/official/vision/beta/modeling/layers/nn_blocks.py b/official/vision/beta/modeling/layers/nn_blocks.py
@@ -714,7 +714,8 @@ def get_config(self):
         'use_depthwise': self._use_depthwise,
         'use_residual': self._use_residual,
         'norm_momentum': self._norm_momentum,
-        'norm_epsilon': self._norm_epsilon
+        'norm_epsilon': self._norm_epsilon,
+        'output_intermediate_endpoints': self._output_intermediate_endpoints
     }
     base_config = super(InvertedBottleneckBlock, self).get_config()
     return dict(list(base_config.items()) + list(config.items()))