-
Notifications
You must be signed in to change notification settings - Fork 3.5k
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
keep parameter names from PyTorch #5887
Conversation
Thanks @t-vi |
@t-vi Thanks for the PR. pytorch_pretrained_bert_uncased.py I think after this PR, the |
What is the error you are seeing? |
Output mismatch between pytorch and tvm. Afer this PR
|
I found the problem and will send a fix. |
#5897 has the fix. |
This patch uses PyTorch parameter names as the name hint in variables.
This means that one can load a stored PyTorch state dict and map the required inputs from that even after conversion.