Converting a model from Pytorch to Tensorflow: Guide to ONNX