Skip to content

Instantly share code, notes, and snippets.

@paulgavrikov

paulgavrikov/medium_onnx_validation.py

Created February 22, 2022 19:05
Show Gist options
  • Save paulgavrikov/d46b67215bf70e5501015f8bdda8503c to your computer and use it in GitHub Desktop.
Save paulgavrikov/d46b67215bf70e5501015f8bdda8503c to your computer and use it in GitHub Desktop.
Download ZIP
Raw
medium_onnx_validation.py
import torch
IN_PATH = ... # path to your .onnx model (OUT_PATH if you used the previous example)
onnx_model = torch.onnx.load(IN_PATH)
torch.onnx.checker.check_model(onnx_model)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment