"vscode:/vscode.git/clone" did not exist on "564d5b76b141d0fa44010a1dc800ed13f7badcbe"
Commit 0940134e authored by Zhichao Lu's avatar Zhichao Lu Committed by pkulzc
Browse files

Set regularize_depthwise to true in mobilenet_v1_argscope.

PiperOrigin-RevId: 192472546
parent b47ca971
......@@ -105,7 +105,8 @@ class SSDMobileNetV1FeatureExtractor(ssd_meta_arch.SSDFeatureExtractor):
with tf.variable_scope('MobilenetV1',
reuse=self._reuse_weights) as scope:
with slim.arg_scope(
mobilenet_v1.mobilenet_v1_arg_scope(is_training=None)):
mobilenet_v1.mobilenet_v1_arg_scope(
is_training=None, regularize_depthwise=True)):
with (slim.arg_scope(self._conv_hyperparams_fn())
if self._override_base_feature_extractor_hyperparams
else context_manager.IdentityContextManager()):
......
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