I am done

This commit is contained in:
2024-10-30 22:14:35 +01:00
parent 720dc28c09
commit 40e2a747cf
36901 changed files with 5011519 additions and 0 deletions

View File

@ -0,0 +1,3 @@
# Copyright (c) ONNX Project Contributors
#
# SPDX-License-Identifier: Apache-2.0

View File

@ -0,0 +1,27 @@
# Copyright (c) ONNX Project Contributors
#
# SPDX-License-Identifier: Apache-2.0
from __future__ import annotations
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)