"cmake/git@developer.sourcefind.cn:OpenDAS/fast_rnnt.git" did not exist on "0818d487b6211d05531a3fd1706f78a46be0acca"
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
...@@ -182,10 +182,14 @@ class FasterRCNNNASFeatureExtractor( ...@@ -182,10 +182,14 @@ 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)):
_, end_points = nasnet.build_nasnet_large( with arg_scope([slim.conv2d,
preprocessed_inputs, num_classes=None, slim.batch_norm,
is_training=self._is_training, slim.separable_conv2d],
final_endpoint='Cell_11') reuse=self._reuse_weights):
_, end_points = nasnet.build_nasnet_large(
preprocessed_inputs, num_classes=None,
is_training=self._is_training,
final_endpoint='Cell_11')
# Note that both 'Cell_10' and 'Cell_11' have equal depth = 2016. # Note that both 'Cell_10' and 'Cell_11' have equal depth = 2016.
rpn_feature_map = tf.concat([end_points['Cell_10'], rpn_feature_map = tf.concat([end_points['Cell_10'],
......
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