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

Remove levit model from ptq conformance tests #3200

Merged
merged 3 commits into from
Jan 22, 2025

Conversation

AlexanderDokuchaev
Copy link
Collaborator

@AlexanderDokuchaev AlexanderDokuchaev commented Jan 20, 2025

Changes

  • Remove levit model_128
  • Fix: no catching exception without message

Reason for changes

  File "/home/jenkins/agent/workspace/NNCF/manual/Daemons/PTQ/ptq_single_model/venv/lib/python3.10/site-packages/timm/models/_builder.py", line 393, in build_model_with_cfg
    load_pretrained(
  File "/home/jenkins/agent/workspace/NNCF/manual/Daemons/PTQ/ptq_single_model/venv/lib/python3.10/site-packages/timm/models/_builder.py", line 193, in load_pretrained
    state_dict = filter_fn(state_dict, model)
  File "/home/jenkins/agent/workspace/NNCF/manual/Daemons/PTQ/ptq_single_model/venv/lib/python3.10/site-packages/timm/models/levit.py", line 707, in checkpoint_filter_fn
    assert 'head' in ka or 'stem.conv1.linear' in ka
AssertionError

@AlexanderDokuchaev AlexanderDokuchaev requested a review from a team as a code owner January 20, 2025 14:08
@github-actions github-actions bot added the NNCF PTQ Pull requests that updates NNCF PTQ label Jan 20, 2025
},
"backends": NNCF_PTQ_BACKENDS,
},
# TODO(AlexanderDokuchaev): Enable after bump timm>1.0.14, https://github.com/huggingface/pytorch-image-models/pull/2412/files
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

When are you going to bump timm version?

@AlexanderDokuchaev AlexanderDokuchaev changed the title Disable levit model for ptq conformance tests Remove levit model from ptq conformance tests Jan 22, 2025
Copy link
Contributor

@alexsu52 alexsu52 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@alexsu52 alexsu52 merged commit bfac8cf into openvinotoolkit:develop Jan 22, 2025
17 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
NNCF PTQ Pull requests that updates NNCF PTQ
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants