Commit Graph

3 Commits (dfb8658100eef51bebe69ca1d8cdd5b88a4dc063)

Author SHA1 Message Date
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