Add model export option using ONNX to facilitate better Netron visualization

[jbloomAus]

https://pytorch.org/docs/stable/onnx.html#example-alexnet-from-pytorch-to-onnx

Not sure how much of a priority this is but it looks cool.