Example of exporting a PyTorch to ONNX format, then performing inference with tract.
Export Model:
python export.pyInference on elephants.jpg:
cargo run
result: Some((22.08386, 102))
Predicts class 102 (tusker).
pytorch-resnet
Folders and files
| Name | Name | Last commit date | ||
|---|---|---|---|---|
parent directory.. | ||||