python中anaconda中tensoeflow-gpu出现错误没有相关模块

报错信息如下

请问一下各位应该如何解决

E:\anaconda3\envs\tensorflow-gpu\python.exe D:/Desktop/FedSim-main/main.py
Traceback (most recent call last):
File "D:/Desktop/FedSim-main/main.py", line 172, in <module>
main()
File "D:/Desktop/FedSim-main/main.py", line 159, in main
options, learner, optimizer = read_options()
File "D:/Desktop/FedSim-main/main.py", line 129, in read_options
mod = importlib.import_module(model_path)
File "E:\anaconda3\envs\tensorflow-gpu\lib\importlib\__init__.py", line 126, in import_module
return _bootstrap._gcd_import(name[level:], package, level)
File "<frozen importlib._bootstrap>", line 994, in _gcd_import
File "<frozen importlib._bootstrap>", line 971, in _find_and_load
File "<frozen importlib._bootstrap>", line 941, in _find_and_load_unlocked
File "<frozen importlib._bootstrap>", line 219, in _call_with_frames_removed
File "<frozen importlib._bootstrap>", line 994, in _gcd_import
File "<frozen importlib._bootstrap>", line 971, in _find_and_load
File "<frozen importlib._bootstrap>", line 953, in _find_and_load_unlocked
ModuleNotFoundError: No module named 'flearn.models.nist.stacked_lstm'

使用不同的模型探索不同的数据集。每个数据集都与一个独特的模型相关。
NIST 的模型在 中mclr定义flearn/models/nist/mclr.py
参考