这个报错之前应该出现过,最近重新跑VIT的时候又出现了,我以为是自己不小心动了代码,查了一下才想起来,numpy的版本不匹配,不知道什么时候重新装上了更高的版本。
具体报错内容为:
Cannot convert a symbolic Tensor (data_augmentation/random_rotation/rotation_matrix/strided_slice:0) to a numpy array. This error may indicate that you're trying to pass a Tensor to a NumPy call, which is not supported
解决办法是在虚拟环境里安装
在输入conda install numpy==1.19.5后发现不行,找不到
于是改成pip install numpy==1.19.5,安装完毕后程序正常
注意:
在我自己查看pycharm数据库时发现numpy已经是1.19.5,但是还有一个numpy-base版本过高,估计就是它的问题