Skip to content

Commit fe0c06d

Browse files
No description
PiperOrigin-RevId: 692325309
1 parent 9084c2e commit fe0c06d

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

tensorflow_examples/lite/model_maker/third_party/efficientdet/efficientdet_arch.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -329,10 +329,11 @@ def build_backbone(features, config):
329329
}
330330
if 'b0' in backbone_name:
331331
override_params['survival_prob'] = 0.0
332-
if config.backbone_config is not None:
332+
backbone_config = config.backbone_config
333+
if backbone_config is not None:
333334
override_params['blocks_args'] = (
334335
efficientnet_builder.BlockDecoder().encode(
335-
config.backbone_config.blocks))
336+
backbone_config.blocks))
336337
override_params['data_format'] = config.data_format
337338
model_builder = backbone_factory.get_model_builder(backbone_name)
338339
_, endpoints = model_builder.build_model_base(

0 commit comments

Comments
 (0)