FedProx源码运行报错:No module named 'flearn.models.mnist.mclr.py'; 'flearn.models.mnist.mclr' is not a package
在pycharm上运行,对main.py进行运行,报错No module named 'flearn.models.nist.stacked_lstm' ,修改dataset默认值为mnist
def read_options():
''' Parse command line arguments or load defaults '''
parser = argparse.ArgumentParser()
parser.add_argument('--optimizer',
help='name of optimizer;',
type=str,
choices=OPTIMIZERS,
default='fedavg')
parser.add_argument('--dataset',
help='name of dataset;',
type=str,
choices=DATASETS,
default='mnist')
parser.add_argument('--model',
help='name of model;',
type=str,
default='mclr.py')
仍然报错ModuleNotFoundError: No module named 'flearn.models.mnist.mclr.py'; 'flearn.models.mnist.mclr' is not a package
引入该数据包相关代码:
# load selected model
if parsed['dataset'].startswith("synthetic"): # all synthetic datasets use the same model
model_path = '%s.%s.%s.%s' % ('flearn', 'models', 'synthetic', parsed['model'])
else:
model_path = '%s.%s.%s.%s' % ('flearn', 'models', parsed['dataset'], parsed['model'])
mod = importlib.import_module(model_path)
learner = getattr(mod, 'Model')
源码文件结构
源码地址:https://github.com/litian96/FedProx
以下答案由GPT-3.5大模型与博主波罗歌共同编写:
根据错误提示,可以看出 flearn.models.mnist.mclr
并不是一个包,所以无法直接引入 mclr.py
。因此需要做如下修改:
在 read_options()
函数中将 model
参数默认值改为 "mclr"
即可,因为 "mclr"
已经在 flearn.models.mnist
模块里了,不需要再加上 .py
。
修改后的代码如下:
def read_options():
''' Parse command line arguments or load defaults '''
parser = argparse.ArgumentParser()
parser.add_argument('--optimizer',
help='name of optimizer;',
type=str,
choices=OPTIMIZERS,
default='fedavg')
parser.add_argument('--dataset',
help='name of dataset;',
type=str,
choices=DATASETS,
default='mnist')
parser.add_argument('--model',
help='name of model;',
type=str,
default='mclr') # 默认值改为 mclr,而不是 mclr.py
return parser.parse_args()
另外,在加载模型时,需要注意修改模型的路径,例如:
# load selected model
if parsed['dataset'].startswith("synthetic"):
model_path = '%s.%s.%s.%s' % ('flearn', 'models', 'synthetic', parsed['model'])
else:
model_path = '%s.%s.%s' % ('flearn', 'models', parsed['model'])
mod = importlib.import_module(model_path)
learner = getattr(mod, 'Model')
这样就可以避免出现 No module named 'flearn.models.mnist.mclr.py'
这样的错误了。
如果我的回答解决了您的问题,请采纳!
可能修改没保存,刷新或者重新打开试试?
该回答引用chatgpt:
可能的解决方法如下:
1检查是否已经正确安装了flearn和其它依赖包。你可以在项目根目录下运行pip install -r requirements.txt来安装所需的依赖包。
2检查flearn/models/mnist目录下是否存在mclr.py模块,或者检查模块名是否正确。如果不存在,请从仓库的原始代码中获取这个模块并将其复制到正确的位置。
3检查flearn/models/mnist目录是否有一个__init__.py文件。这个文件用来标记一个目录为Python包,如果这个文件不存在,Python可能无法正确地加载这个包。
4检查flearn/models/mnist目录是否在Python的搜索路径中。你可以使用以下代码来检查Python的搜索路径:import sys; print(sys.path)。
如果你确定上面回答的原因都排除了,那清除一下缓存吧,再运行,说不定就OK了。
python -m compileall .
No module named ‘models‘ 解决方法(亲测)
可以借鉴下
https://blog.csdn.net/OnlyHeart_/article/details/124935824