Assign requires shapes of both tensors to match. lhs shape= [4096,1470] rhs shape= [4096,539]

• ​​报错如下​​
• ​​解决方法​​

报错如下

解决方法

笔者报错是由于训练类别与测试类别的配置参数不对应！

• 将下列代码：

• 改为(自己训练的数据类别)：

• 即可！