0f1b9f861ebec4cd6811f1e7fa13e63854f429ca,python/src/nnabla/utils/converter/commands.py,,_import_file,#Any#Any#,28

Before Change

def _import_file(args, ifiles):
    if len(ifiles) == 1 and os.path.splitext(ifiles[0])[1] == ".nnp":
        args.import_format = "NNP"
    if len(ifiles) == 1 and os.path.splitext(ifiles[0])[1] == ".onnx":
        args.import_format = "ONNX"
    if args.import_format == "NNP":
        // Input file that has unsupported extension store into output nnp

After Change

def _import_file(args, ifiles):
    if len(ifiles) == 1:
        ext = os.path.splitext(ifiles[0])[1]
        if ext == ".nnp":
            args.import_format = "NNP"
        elif ext == ".onnx":
            args.import_format = "ONNX"
        elif ext == ".pb":
            args.import_format = "TF_PB"
        elif ext == ".ckpt":
            args.import_format = "TF_CKPT"

    if args.import_format == "NNP":
        // Input file that has unsupported extension store into output nnp
        // archive or directory.
        return NnpImporter(*ifiles,

Instances