-
Notifications
You must be signed in to change notification settings - Fork 705
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
Update for nncf_task #145
Update for nncf_task #145
Conversation
a9c44f3
to
5995978
Compare
5995978
to
79edc56
Compare
646dc61
to
1e2267a
Compare
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
@AlexanderDokuchaev, Thank you for making this PR. I have few comments and opens that I left in the code.
anomalib/integration/__init__.py
Outdated
"""Integration.""" | ||
|
||
# Copyright (C) 2021 Intel Corporation | ||
# | ||
# Licensed under the Apache License, Version 2.0 (the "License"); | ||
# you may not use this file except in compliance with the License. | ||
# You may obtain a copy of the License at | ||
# | ||
# http://www.apache.org/licenses/LICENSE-2.0 | ||
# | ||
# Unless required by applicable law or agreed to in writing, | ||
# software distributed under the License is distributed on an "AS IS" BASIS, | ||
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | ||
# See the License for the specific language governing permissions | ||
# and limitations under the License. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Rather than creating anomalib/integration
, we could place nncf_callback
and utils.py
into anomalib/utils/callbacks/nncf
. In this case, the existing flow would be kept as is. More specifically, the structure of anomailb/utils/callback
would be something like the following:
.
├── __init__.py
├── callbacks
│ ├── __init__.py
│ ├── cdf_normalization.py
│ ├── compress.py
│ ├── min_max_normalization.py
│ ├── model_loader.py
│ ├── nncf
│ │ ├── __init__.py
│ │ ├── nncf_callback.py
│ │ └── utils.py
│ ├── save_to_csv.py
│ ├── tiler.py
│ ├── timer.py
│ └── visualizer_callback.py
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I agree. Let's stick to the existing structure and place the nncf files under callbacks.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
done
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks @AlexanderDokuchaev! Really appreciate the effort!
Rework nncf integration to use in OTE:
compression
toopenvino
in config andCompressModelCallback
toOpenVINOCalback
because model_optimizer does not compress model and it can confuse users.