Dynamic shape - Githubissues

microsoft / nni

An open source AutoML toolkit for automate machine learning lifecycle, including feature engineering, neural architecture search, model compression and hyper-parameter tuning.

https://nni.readthedocs.io

MIT License

14k stars 1.81k forks source link

Dynamic shape #5625

Open BillAmihom opened 1 year ago

BillAmihom commented 1 year ago

Description

This commit is to support the situation that the input shape and output shapeof models are dynamic. It concludes procession torch2onnx & onnx2trt .

Test Options

[ ] fast test
[ ] full test - HPO
[ ] full test - NAS
[ ] full test - compression

Checklist

[ ] test case
[ ] doc

microsoft / nni

Dynamic shape #5625

Description

Test Options

Checklist

How to test