我具有以下文件结构：
mytests
models
parts
__init__.py
...
__init__.py
model.py
test.py
这是以下内容__init__.py：
from parts import *
from models import My_transformer
if __name__=='__main__':
# ...
这是以下内容model.py：
from parts.attention import Pre_Net
class My_transformer(tf.keras.Model):
def __init__(self,config,logger=None):
super(My_transformer, self).__init__()
...
运行时test.py，出现以下错误：
/mytests/models/__init__.py in <module>()
----> 1 from parts import *
ModuleNotFoundError: No module named 'parts'
在这一行：
from models import My_transformer
如何解决此错误？我从Jupyter Notebook运行test.py。
更新资料
我在__init__.py里面parts，看起来如下：
from attention import *
if __name__=='__main__':
print('ok')
您应该具有以下结构和导入：
mytests
models
parts
__init__.py
attention.py
__init__.py
model.py
test.py
models/parts/__init__.py 内容：
from .attention import Pre_Net
# or from .attention import *
models/parts/attention.py 内容：
class Pre_Net:
pass
models/__init__.py 内容：
from .model import *
models/model.py 内容：
from models.parts import Pre_Net
class My_transformer(tf.keras.Model):
def __init__(self,config,logger=None):
super(My_transformer, self).__init__()
...
test.py 内容：
from models import My_transformer
本文收集自互联网，转载请注明来源。
如有侵权，请联系 [email protected] 删除。
我来说两句