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

Refactor NNCF compression statistics reporting API #688

Merged
merged 14 commits into from
May 24, 2021
Prev Previous commit
Next Next commit
Fix pylint
  • Loading branch information
andrey-churkin committed May 23, 2021
commit 3498268d7009f07c851f05e2689b2758e2c89738
1 change: 0 additions & 1 deletion beta/nncf/tensorflow/pruning/base_algorithm.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,6 @@
limitations under the License.
"""

from typing import Dict
from typing import List
from typing import Tuple

Expand Down
4 changes: 2 additions & 2 deletions nncf/torch/pruning/base_algo.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,10 +13,10 @@

from functools import partial
from functools import update_wrapper
from texttable import Texttable
from torch import nn
from typing import List

from torch import nn

from nncf.torch.algo_selector import ZeroCompressionLoss
from nncf.common.graph.transformations.commands import TargetType
from nncf.torch.compression_method_api import PTCompressionAlgorithmBuilder
Expand Down