您好,我修改了main_alexnet.py,用到自己的数据集训练,结果报错
lyt@lyt-Default-string:~/tensorflow_alexnet_classify-master$ python3 main_alexnet.py
Traceback (most recent call last):
File "main_alexnet.py", line 12, in
from datagenerator import ImageDataGenerator
File "/home/lyt/tensorflow_alexnet_classify-master/datagenerator.py", line 7, in
from tensorflow.data import Dataset
ImportError: No module named 'tensorflow.data'
您好,我修改了main_alexnet.py,用到自己的数据集训练,结果报错
lyt@lyt-Default-string:~/tensorflow_alexnet_classify-master$ python3 main_alexnet.py
Traceback (most recent call last):
File "main_alexnet.py", line 12, in
from datagenerator import ImageDataGenerator
File "/home/lyt/tensorflow_alexnet_classify-master/datagenerator.py", line 7, in
from tensorflow.data import Dataset
ImportError: No module named 'tensorflow.data'