tensorflow error : Assign requires shapes of both tensors to match.

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

训练和测试中的网络参数不同导致的错误。

但我使用的是同一个model文件,最后发现256和8192之间正好差了一个batch = 32 的倍数,应该是训练网络中,参数写错了。

原训练网络:

……
batch = 32
……
x = tf.placeholder(tf.float32, shape=[1, IMG_W, IMG_H, 3])
logit = model.inference(x, BATCH_SIZE, N_ClASSES) 
……
……

将batch = 1后,错误解决了。

猜你喜欢

转载自blog.csdn.net/i96jie/article/details/82461428