model_tools 源码

PyTorch模型转换工具 从PyTorch转换为ONNX和TorchScript 支持源类型: 火炬检查站 火炬模型 火炬脚本 依赖关系: 火炬 用法: torch2onnx.py [-h] [--src-model-path SRC_MODEL_PATH] [--dst-model-dir DST_MODEL_DIR] [--dst-model-name DST_MODEL_NAME] [-o ONNX] [-t TORCH_SCRIPT] [-f NET_FILE] [-c CLS_NAME] [--func-name FUNC_NAME] [-b BATCH] [--input-name INPUT_NAME] [--input-shape INPUT_SHAPE [INPUT_SHAPE ...]] [--output-name