-
Notifications
You must be signed in to change notification settings - Fork 1.8k
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
QuickStart example for pruning is not working properly !! #5768
Comments
Same problem here |
same |
using pytorch version 2.1. it works for me |
|
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Describe the bug:
QuickStart Example provides Error at model SpeedUp. AttributeError: 'NoneType' object has no attribute 'startswith'. I tried running the pruning example quick start code and all worked fine until the section of model speedup.
Environment:
Reproduce the problem
from nni.compression.speedup import ModelSpeedup
ModelSpeedup(model, torch.rand(3, 1, 28, 28).to(device), masks).speedup_model()
AttributeError Traceback (most recent call last)
in <cell line: 7>()
5 from nni.compression.speedup import ModelSpeedup
6
----> 7 ModelSpeedup(model, torch.rand(3, 1, 28, 28).to(device), masks).speedup_model()
2 frames
/usr/local/lib/python3.10/dist-packages/nni/common/concrete_trace_utils/concrete_tracer.py in trace(self, root, autowrap_modules, autowrap_leaf_function, autowrap_leaf_class, leaf_module, fake_middle_class, concrete_args, use_operator_patch, operator_patch_backlist, forward_function_name)
956 elif func.name != func.qualname and func.qualname != 'boolean_dispatch..fn':
957 # method
--> 958 if func.module.startswith('_') and func.module != 'main':
959 path = sys.modules[func.module[1:]]
960 else:
AttributeError: 'NoneType' object has no attribute 'startswith'
The text was updated successfully, but these errors were encountered: