Skip to content

GitLab

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(
resnet_v1_base_model_name,
first_stage_features_stride,
conv_hyperparams,
batch_norm_trainable=False,
batch_norm_trainable=True,
pad_to_multiple=32,
weight_decay=0.0,
fpn_min_level=2,
......@@ -308,7 +308,7 @@ class FasterRCNNResnet50FpnKerasFeatureExtractor(
def __init__(self,
is_training,
first_stage_features_stride=16,
batch_norm_trainable=False,
batch_norm_trainable=True,
conv_hyperparams=None,
weight_decay=0.0,
fpn_min_level=2,
......@@ -348,7 +348,7 @@ class FasterRCNNResnet101FpnKerasFeatureExtractor(
def __init__(self,
is_training,
first_stage_features_stride=16,
batch_norm_trainable=False,
batch_norm_trainable=True,
conv_hyperparams=None,
weight_decay=0.0,
fpn_min_level=2,
......@@ -389,7 +389,7 @@ class FasterRCNNResnet152FpnKerasFeatureExtractor(
def __init__(self,
is_training,
first_stage_features_stride=16,
batch_norm_trainable=False,
batch_norm_trainable=True,
conv_hyperparams=None,
weight_decay=0.0,
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