InvalidArgumentError (see above for traceback): Inputs to operation gradients_1/AddN_39 of type _MklAddN must have the same size and shape. Input 0: [30,7,7,128] != input 1: [188160]
[[Node: gradients_1/AddN_39 = _MklAddN[N=2, T=DT_FLOAT, _kernel="MklOp", _device="/job:localhost/replica:0/task:0/device:CPU:0"](gradients_1/MobileFaceNet/MobileFaceNet/Conv2d_6_InvResBlock_1/Add_grad/tuple/control_dependency_1, gradients_1/MobileFaceNet/MobileFaceNet/Conv2d_6_InvResBlock_1/Conv/Conv2D_grad/tuple/control_dependency, gradients_1/MobileFaceNet/MobileFaceNet/Conv2d_6_InvResBlock_1/Add_grad/tuple/control_dependency_1:1, gradients_1/MobileFaceNet/MobileFaceNet/Conv2d_6_InvResBlock_1/Conv/Conv2D_grad/tuple/control_dependency:1)]]
我使用的tensorflow版本为1.9.0,报上面错误,排查半天以为损失函数写错了。
结果,最后,换了同事的1.8.0版本,完美运行。至今不知道啥毛病。