search

NeVerTools / pyNeVer

A Python library for learning and verification of neural networks and other machine learning models
Other
9 stars 3 forks source link

Improve PyTorch conversion #2

Closed sdemarch closed 1 year ago

sdemarch commented 2 years ago

Improve PyTorch to ONNX conversion by asking for the input shape of the network and creating node identifiers in the process.