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

microsoft / nni

An open source AutoML toolkit for automate machine learning lifecycle, including feature engineering, neural architecture search, model compression and hyper-parameter tuning.

https://nni.readthedocs.io

MIT License

13.88k stars 1.81k forks source link

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

Open mo-shahab opened 4 months ago

mo-shahab commented 4 months ago

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?:

mo-shahab commented 4 months ago

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