mindspore-mindinsight/mindinsight/mindconverter/tools

• Tool Tutorial
• • Fix CheckPoint file Tool
  • • Overview
    • Usage
    • Example
    • Limitation

Tool Tutorial

查看中文

• Tool Tutorial
  • Fix CheckPoint file Tool
    • Overview
    • Usage
    • Example
    • Limitation

Fix CheckPoint file Tool

Overview

Requirements

MindSpore>=1.2.0

User may need to change module name or variable name in order to improve the readability of converted MindSpore script. However, these changes will result in failing to load checkpoint because of unmatched weight name. The Fix CheckPoint file Tool is used to fix weight names in checkpoint file.

Usage

usage: fix_checkpoint_file.py [-h]
                              source_py_file fixed_py_file source_ckpt_file
                              fixed_ckpt_file

Fix weight name in CheckPoint file.

positional arguments:
  source_py_file    source model script file
  fixed_py_file     fixed model script file
  source_ckpt_file  source_checkpoint file  
  fixed_ckpt_file   fixed_checkpoint file

optional arguments:
  -h, --help        show this hekp message and exit

Example

Assuming that the source model script file is xxx/model.py, the fixed model script file is xxx/fixed_model.py, the source checkpoint file is xxx/model.ckpt and the new checkpoint file is xxx/fixed_model.ckpt.

The command is that:

python -m mindinsight.mindconverter.tools.fix_checkpoit_file xxx/model.py xxx/fixed_model.py xxx/model.ckpt xxx/fixed_model.ckpt

If generation is successful, the result below would be shown:

Saved new checkpoint file to xxx/fixed_model.ckpt.

Limitation

1. Only MindSpore Script and CheckPoint file generated by MindConverter using graph-based conversion (using --model_file) are supported in the tool.
2. The situation that only variable name or class name in scripts has changed are supported, while the one that model structure or script structure (add or delete operator) has changed is unsupported.
3. MindSpore is required by the tool, so make sure MindSpore installed correctly.