TF的环境:
Windows10
Anaconda3
Tensowflow 2.5
Python 3.8
用model_main_tf2.py训练模型,出现题目中这个错误,查找了一圈,有几个方法:
1.降低numpy的版本到1.95
2.降低Python的版本到3.6
有人说这两种方法都可以成功,可以试试。
但是我觉得这两种方法不是真正的解决问题,所以继续找答案。
找到这个方法:
找到array_ops.py,我的在
D:\ProgramData\Anaconda3\envs\tensorflow\Lib\site-packages\tensorflow\python\ops\array_ops.py
根据错误提示,可以找到这个文件位置。
在文件中,增加
from tensorflow.python.ops.math_ops import reduce_prod
然后找到
def _constant_if_small(value, shape, dtype, name)
定义,
原代码是这样的:
def