Spaces:
Running
Running
File size: 684 Bytes
dc2106c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
# Copyright (c) ONNX Project Contributors
#
# SPDX-License-Identifier: Apache-2.0
import argparse
from onnx import NodeProto, checker, load
def check_model() -> None:
parser = argparse.ArgumentParser("check-model")
parser.add_argument("model_pb", type=argparse.FileType("rb"))
args = parser.parse_args()
model = load(args.model_pb)
checker.check_model(model)
def check_node() -> None:
parser = argparse.ArgumentParser("check-node")
parser.add_argument("node_pb", type=argparse.FileType("rb"))
args = parser.parse_args()
node = NodeProto()
node.ParseFromString(args.node_pb.read())
checker.check_node(node)
|