ssp4onnx

Simple Split for ONNX. A simple tool that automatically splits ONNX models of specified weight sizes.

Usage

pip install ssp4onnx

ssp4onnx -i model.onnx --auto_split_max_size 1GB

Options:

• -i/--input_onnx_file: input ONNX file (required)
• -o/--output_dir: output directory (optional, default: same directory as input)
• -s/--auto_split_max_size: target partition size, supports KB, MB, GB (default: 100MB)

Sample reuslts

Split.1 (1.1GB)	Split.2 (1.1GB)	Split.3 (0.4GB)