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