Skip to content

GitLab

Commit 6f521d91 authored by A. Unique TensorFlower's avatar A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 481247611
parent 45038543
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/vision/modeling/backbones/spinenet.py
View file @ 6f521d91
...@@ -554,7 +554,7 @@ def build_spinenet( ...@@ -554,7 +554,7 @@ def build_spinenet(
assert backbone_type == 'spinenet', (f'Inconsistent backbone type ' assert backbone_type == 'spinenet', (f'Inconsistent backbone type '
f'{backbone_type}') f'{backbone_type}')
model_id = backbone_cfg.model_id model_id = str(backbone_cfg.model_id)
if model_id not in SCALING_MAP: if model_id not in SCALING_MAP:
raise ValueError( raise ValueError(
'SpineNet-{} is not a valid architecture.'.format(model_id)) 'SpineNet-{} is not a valid architecture.'.format(model_id))
......
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