1. 程式人生 > >tensorflow error : Assign requires shapes of both tensors to match.

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後,錯誤解決了。