Commit Graph

4 Commits (a667d3d8f0f13b0759dc919246a130b6a9fc9772)

Author SHA1 Message Date
Ross Wightman ba641e07ae Add support for dynamo based onnx export 2024-03-13 12:05:26 -07:00
Youssef Boulaoaune a9c29a05de add input_size & batch_size to onnx_export 2023-09-21 14:46:25 -07:00
Ross Wightman 38c474e3de Update README, add --reparm to onnx_export 2023-08-25 11:04:34 -07:00
Ross Wightman aef6e562e4 Add onnx utils and export code, tweak padding and conv2d_same for better dynamic export with recent PyTorch 2023-04-11 17:03:57 -07:00