onnx / tutorials

Tutorials for creating and using ONNX models
Apache License 2.0
3.37k stars 629 forks source link

Pytorch to Tensorflow conversion #153

Open adrianosantospb opened 5 years ago

adrianosantospb commented 5 years ago

I have a question about trained model conversion. I have a PyTouch model trained and I'd like to convert to Tensorflow model. I found this example but I don't understand why is necessary to create something like this to convert the model?

dummy_input = Variable(torch.randn(1, 1, 28, 28)) ...

Tks for your help.