diff --git a/model_zoo/official/cv/faster_rcnn/src/convert_checkpoint.py b/model_zoo/official/cv/faster_rcnn/src/convert_checkpoint.py index 1cc449734c..dab0974e2c 100644 --- a/model_zoo/official/cv/faster_rcnn/src/convert_checkpoint.py +++ b/model_zoo/official/cv/faster_rcnn/src/convert_checkpoint.py @@ -44,7 +44,7 @@ def load_weights(model_path, use_fp16_weight): param_name = msname if "down_sample_layer.0" in param_name: param_name = param_name.replace("down_sample_layer.0", "conv_down_sample") - if "down_sample-layer.1" in param_name: + if "down_sample_layer.1" in param_name: param_name = param_name.replace("down_sample_layer.1", "bn_down_sample") weights[param_name] = ms_ckpt[msname].data.asnumpy() if use_fp16_weight: @@ -60,5 +60,5 @@ def load_weights(model_path, use_fp16_weight): return param_list if __name__ == "__main__": - parameter_list = load_weights(args_opt.ckpt_file, use_fp16_weight=True) + parameter_list = load_weights(args_opt.ckpt_file, use_fp16_weight=False) save_checkpoint(parameter_list, "resnet50_backbone.ckpt") diff --git a/model_zoo/official/cv/maskrcnn/src/convert_checkpoint.py b/model_zoo/official/cv/maskrcnn/src/convert_checkpoint.py index 1cc449734c..c3600a58a6 100644 --- a/model_zoo/official/cv/maskrcnn/src/convert_checkpoint.py +++ b/model_zoo/official/cv/maskrcnn/src/convert_checkpoint.py @@ -44,7 +44,7 @@ def load_weights(model_path, use_fp16_weight): param_name = msname if "down_sample_layer.0" in param_name: param_name = param_name.replace("down_sample_layer.0", "conv_down_sample") - if "down_sample-layer.1" in param_name: + if "down_sample_layer.1" in param_name: param_name = param_name.replace("down_sample_layer.1", "bn_down_sample") weights[param_name] = ms_ckpt[msname].data.asnumpy() if use_fp16_weight: