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.

Sign up for GitHub

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

Jump to bottom

ModuleNotFoundError: No module named 'nni.algorithms.compression' #5750

Open

mo-shahab opened this issue Mar 5, 2024 · 1 comment

mo-shahab commented Mar 5, 2024

Describe the issue:

Environment:

NNI version: 3.0
Training service (local|remote|pai|aml|etc):
Client OS:
Server OS (for remote mode only):
Python version: 3.12
PyTorch/TensorFlow version: Pytorch=2.2.1
Is conda/virtualenv/venv used?: yes
Is running in Docker?: no

Configuration:

Experiment config (remember to remove secrets!):
Search space:

Log message:

nnimanager.log:
dispatcher.log:
nnictl stdout and stderr:

How to reproduce it?:

Author

mo-shahab commented Mar 5, 2024 •

edited

import statements in the script

import functools, copy
import numpy as np
import torch
import torch.nn as nn
import torch.nn.functional as F
from nni.algorithms.compression.pytorch.pruning import L1FilterPruner, L1FilterPrunerMasker

from nni.algorithms.compression.pytorch.pruning import L1FilterPruner, L1FilterPrunerMasker

loc giving the error

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment