Option to save models with torchscript and/or ONNX

jcreinhold / tiramisu-brulee

A 2D and 3D PyTorch implementation of the Tiramisu CNN

Other

5 stars 2 forks source link

Open jcreinhold opened 3 years ago

jcreinhold commented 3 years ago

Add an option to save models with torchscript and/or ONNX

Improve interoperability between tools by optionally saving models w/ torchscript and/or ONNX

jcreinhold commented 2 years ago

Added ONNX converter: tiramisu_brulee/experiment/cli/to_onnx.py