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

Mkaglins/pruning masks applying #985

Prev Previous commit
Next Next commit
Update nncf/torch/module_operations.py with docs
Co-authored-by: Daniil Lyakhov <daniil.lyakhov@intel.com>
  • Loading branch information
mkaglins and daniil-lyakhov committed Oct 19, 2021
commit 351ccb84d9c0f922b4eeace3ac4aab9cd95c7c97
5 changes: 5 additions & 0 deletions nncf/torch/module_operations.py
Original file line number Diff line number Diff line change
Expand Up @@ -53,6 +53,11 @@ def __init__(self, op):


class UpdateParameterList(BaseOp):
"""
A module which updates attributes of a module fed to
forward method call by operand call.
"""

def __init__(self, param_names, op):
super().__init__(op)
self._param_names = param_names
Expand Down