Commit 312b09e2 authored by Zhichao Lu's avatar Zhichao Lu Committed by pkulzc
Browse files

Use keyword arguments in model builder for ssd feature extractor. This was...

Use keyword arguments in model builder for ssd feature extractor. This was incorreclty setting reuse argument.

PiperOrigin-RevId: 190940462
parent 5bbe41c4
......@@ -159,8 +159,9 @@ def _build_ssd_model(ssd_config, is_training, add_summaries):
# Feature extractor
feature_extractor = _build_ssd_feature_extractor(
ssd_config.feature_extractor, is_training,
ssd_config.inplace_batchnorm_update)
feature_extractor_config=ssd_config.feature_extractor,
is_training=is_training,
inplace_batchnorm_update=ssd_config.inplace_batchnorm_update)
box_coder = box_coder_builder.build(ssd_config.box_coder)
matcher = matcher_builder.build(ssd_config.matcher)
......
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