From ce240376adbe422336323795d815cd7f9c49aba0 Mon Sep 17 00:00:00 2001 From: Hiroshi Yoshioka <40815708+hyoshioka0128@users.noreply.github.com> Date: Thu, 30 May 2024 00:41:16 +0900 Subject: [PATCH] =?UTF-8?q?Update=20pytorch=5Fparser.py=20(Typo=20"Pytorch?= =?UTF-8?q?"=E2=86=92"PyTorch")?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit https://github.com/microsoft/MMdnn/blob/master/mmdnn/conversion/pytorch/pytorch_parser.py #PingMSFTDocs --- mmdnn/conversion/pytorch/pytorch_parser.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mmdnn/conversion/pytorch/pytorch_parser.py b/mmdnn/conversion/pytorch/pytorch_parser.py index 7ee90f5e..3e74f703 100644 --- a/mmdnn/conversion/pytorch/pytorch_parser.py +++ b/mmdnn/conversion/pytorch/pytorch_parser.py @@ -72,7 +72,7 @@ def get_weight_name(self, node): def __init__(self, model_file_name, input_shape): super(PytorchParser, self).__init__() if not os.path.exists(model_file_name): - print("Pytorch model file [{}] is not found.".format(model_file_name)) + print("PyTorch model file [{}] is not found.".format(model_file_name)) assert False # test @@ -128,7 +128,7 @@ def _set_output_shape(self, source_node, IR_node): new_dim = shape.dim.add() if not shape_pytorch: - print("Warning: Pytorch cannot inference outputshape of \"{}\" with operator \"{}\". Setting outputshape manually in json file is alternative .".format(source_node.name, source_node.type)) + print("Warning: PyTorch cannot inference outputshape of \"{}\" with operator \"{}\". Setting outputshape manually in json file is alternative .".format(source_node.name, source_node.type)) IR_node.attr["_output_shapes"].list.shape.extend([shape]) return