Fix syntax error of config when upgrading model version (#4584)

Comment the first line of meta_info['config'] to avoid syntax error
pull/4526/head^2
CescMessi 4 years ago committed by GitHub
parent 3620bb7489
commit 340d6eb9bb
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
  1. 2
      tools/model_converters/upgrade_model_version.py

@ -123,7 +123,7 @@ def convert(in_file, out_file, num_classes):
out_state_dict = OrderedDict()
meta_info = checkpoint['meta']
is_two_stage, is_ssd, is_retina, reg_cls_agnostic = parse_config(
meta_info['config'])
'#' + meta_info['config'])
if meta_info['mmdet_version'] <= '0.5.3' and is_retina:
upgrade_retina = True
else:

Loading…
Cancel
Save