Skip to content

GitLab

Commit a4ffb34d authored by Zhichao Lu's avatar Zhichao Lu Committed by lzc5123016
Browse files

Make sure that reuse is passed to the network. 

PiperOrigin-RevId: 187677107
parent 3fff4b86
Show whitespace changes
Inline Side-by-side
Showing with 8 additions and 4 deletions
research/object_detection/models/faster_rcnn_nas_feature_extractor.py
View file @ a4ffb34d
...@@ -182,6 +182,10 @@ class FasterRCNNNASFeatureExtractor( ...@@ -182,6 +182,10 @@ class FasterRCNNNASFeatureExtractor(
with slim.arg_scope(nasnet_large_arg_scope_for_detection( with slim.arg_scope(nasnet_large_arg_scope_for_detection(
is_batch_norm_training=self._train_batch_norm)): is_batch_norm_training=self._train_batch_norm)):
with arg_scope([slim.conv2d,
slim.batch_norm,
slim.separable_conv2d],
reuse=self._reuse_weights):
_, end_points = nasnet.build_nasnet_large( _, end_points = nasnet.build_nasnet_large(
preprocessed_inputs, num_classes=None, preprocessed_inputs, num_classes=None,
is_training=self._is_training, is_training=self._is_training,
......
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