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.

Sign up for GitHub

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

Jump to bottom

--disable_jit_freeze flag is not working in many Pytorch models #153

Open
kkontny opened this issue Oct 10, 2022 · 2 comments
Open

--disable_jit_freeze flag is not working in many Pytorch models #153

kkontny opened this issue Oct 10, 2022 · 2 comments
Assignees
@MarcelWilnicki

Comments

@kkontny
Copy link
Contributor

kkontny commented Oct 10, 2022

In some models (I found it in resnet50, vgg16, bert_base, roberta, but please check other too) the disable_jit_freeze parameter isn't correctly passed to the run_pytorch_fp function.
It makes tests with jit freeze not working. It is very useful testing the eager mode.
@jan-grzybek-ampere
Copy link
Member

@MarcelWilnicki please work on this "in the background" while giving way to more time-sensitive stuff

@MarcelWilnicki
Copy link
Collaborator

Ok, I will take a look at it.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees

@MarcelWilnicki MarcelWilnicki

Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
3 participants
@MarcelWilnicki @jan-grzybek-ampere @kkontny