Skip to content

GitLab

"...composable_kernel.git" did not exist on "d714fa15cb556fcb5f132f1a0c7cfbc22163a253"
Commit 3e73c76c authored by syiming's avatar syiming
Browse files

change batchnorm trainable to true

parent 145ac875
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
research/object_detection/models/faster_rcnn_resnet_v1_fpn_keras_feature_extractor.py
View file @ 3e73c76c
...@@ -113,7 +113,7 @@ class FasterRCNNResnetV1FpnKerasFeatureExtractor( ...@@ -113,7 +113,7 @@ class FasterRCNNResnetV1FpnKerasFeatureExtractor(
resnet_v1_base_model_name, resnet_v1_base_model_name,
first_stage_features_stride, first_stage_features_stride,
conv_hyperparams, conv_hyperparams,
batch_norm_trainable=False, batch_norm_trainable=True,
pad_to_multiple=32, pad_to_multiple=32,
weight_decay=0.0, weight_decay=0.0,
fpn_min_level=2, fpn_min_level=2,
...@@ -308,7 +308,7 @@ class FasterRCNNResnet50FpnKerasFeatureExtractor( ...@@ -308,7 +308,7 @@ class FasterRCNNResnet50FpnKerasFeatureExtractor(
def __init__(self, def __init__(self,
is_training, is_training,
first_stage_features_stride=16, first_stage_features_stride=16,
batch_norm_trainable=False, batch_norm_trainable=True,
conv_hyperparams=None, conv_hyperparams=None,
weight_decay=0.0, weight_decay=0.0,
fpn_min_level=2, fpn_min_level=2,
...@@ -348,7 +348,7 @@ class FasterRCNNResnet101FpnKerasFeatureExtractor( ...@@ -348,7 +348,7 @@ class FasterRCNNResnet101FpnKerasFeatureExtractor(
def __init__(self, def __init__(self,
is_training, is_training,
first_stage_features_stride=16, first_stage_features_stride=16,
batch_norm_trainable=False, batch_norm_trainable=True,
conv_hyperparams=None, conv_hyperparams=None,
weight_decay=0.0, weight_decay=0.0,
fpn_min_level=2, fpn_min_level=2,
...@@ -389,7 +389,7 @@ class FasterRCNNResnet152FpnKerasFeatureExtractor( ...@@ -389,7 +389,7 @@ class FasterRCNNResnet152FpnKerasFeatureExtractor(
def __init__(self, def __init__(self,
is_training, is_training,
first_stage_features_stride=16, first_stage_features_stride=16,
batch_norm_trainable=False, batch_norm_trainable=True,
conv_hyperparams=None, conv_hyperparams=None,
weight_decay=0.0, weight_decay=0.0,
fpn_min_level=2, fpn_min_level=2,
......
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