Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

After converting my PyTorch model to Tensorflow, the output prediction changes #243

Open
laurence-lin opened this issue May 17, 2021 · 1 comment
Labels

Comments

@laurence-lin
Copy link

I'd got a dog image, and I attempt to use a Pytorch model to classify, then convert it to ONNX and to Tensorflow type.

I expect to get a same classification result, since it's originally the same model. However, the Pytorch model outputs "tennis_ball" class while Tensorflow model outputs "Samoyed" class.

Based on the true label, I suppose the Tensorflow model is more accurate, but that result confused me. Why would the same model perform differently?

How could I do to make it static for the model performance?

Here is my working process: https://colab.research.google.com/drive/1pBU_pYdmBHPEPf2C5fRTGkP_cs5kxL0e?usp=sharing

@yuslepukhin
Copy link

You might get a better response at github.com/pytorch/pytorch

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Projects
None yet
Development

No branches or pull requests

2 participants