Skip to content

v0.2.0 - 7 New Features: FLOPs, Diff, Extract, Simplify, Report, Rename & More

Latest

Choose a tag to compare

@Luv-Goel Luv-Goel released this 10 May 05:00
· 5 commits to main since this release

v0.2.0

7 new CLI commands and a major under-the-hood cleanup:

New Features

  • *\onnx-surgery flops * — Estimate MACs, FLOPs, total parameters, and memory footprint with per-operator ASCII bar chart
  • *\onnx-surgery diff * — Structural comparison of two models: added/removed/changed nodes, input/output diffs, op distribution shifts
  • *\onnx-surgery extract --from X --to Y* — Slice a model between named tensors into a standalone .onnx file
  • *\onnx-surgery simplify * — Graph simplification: constant folding, Identity removal, optional BN-Conv fusion
  • *\onnx-surgery report -o report.html* — Self-contained HTML report with summary cards, node tables, FLOPs breakdown
  • *\onnx-surgery rename --map old:new* — Bulk rename tensors throughout the entire model
  • *\onnx-surgery info --shapes * — Enhanced info with computational estimates

Fixes

  • All unicode/emoji replaced with ASCII for Windows cp1252 terminal compatibility
  • 100% test pass rate across 12 tests on Python 3.10/3.11/3.12
  • CLI now has 13 subcommands total