From 712b94e7050d2bfb1f9897f2d507f10b34079478 Mon Sep 17 00:00:00 2001 From: Gary Miguel Date: Tue, 17 May 2022 12:53:06 -0700 Subject: [PATCH] Updated PyTorch ONNX exporter (markdown) --- PyTorch-ONNX-exporter.md | 36 +++++++++++++++++------------------- 1 file changed, 17 insertions(+), 19 deletions(-) diff --git a/PyTorch-ONNX-exporter.md b/PyTorch-ONNX-exporter.md index f8a25af..20fa9ef 100644 --- a/PyTorch-ONNX-exporter.md +++ b/PyTorch-ONNX-exporter.md @@ -4,24 +4,22 @@ Documentation for developing the PyTorch-ONNX exporter (`torch.onnx`). -- [Table of Contents](#table-of-contents) -- [Development process](#development-process) - - [Environment setup](#environment-setup) - - [Fork PyTorch](#fork-pytorch) - - [Build PyTorch](#build-pytorch) - - [Optional build tips](#optional-build-tips) - - [Install additional dependencies](#install-additional-dependencies) - - [ONNX Runtime](#onnx-runtime) - - [ONNX](#onnx) - - [TorchVision](#torchvision) - - [Sanity check](#sanity-check) - - [VS Code](#vs-code) - - [Pull requests](#pull-requests) - - [Tests](#tests) -- [Links](#links) - - [Relevant parts of PyTorch repo](#relevant-parts-of-pytorch-repo) -- [Features](#features) - - [Quantized model export](#quantized-model-export) +* [Development process](#development-process) + * [Environment setup](#environment-setup) + * [Fork PyTorch](#fork-pytorch) + * [Build PyTorch](#build-pytorch) + * [Optional build tips](#optional-build-tips) + * [Install additional dependencies](#install-additional-dependencies) + * [ONNX and ONNX Runtime](#onnx-and-onnx-runtime) + * [TorchVision](#torchvision) + * [Sanity check](#sanity-check) + * [VS Code](#vs-code) + * [Pull requests](#pull-requests) + * [Tests](#tests) +* [Links](#links) + * [Relevant parts of PyTorch repo](#relevant-parts-of-pytorch-repo) +* [Features](#features) + * [Quantized model export](#quantized-model-export) # Development process @@ -101,7 +99,7 @@ The ONNX tests depend on torchvision. This is tricky because TorchVision depends on PyTorch, but we don't want our package manager to install PyTorch, we want to use our locally built one. The best solution I've found is to install torchvision with pip (so that -conda doesn't try to manage it) without any deps (so that pip doesn't install +conda doesn't try to manage it) without any deps (so that pip doesn't install pytorch and the locally built version is used). ```sh