Graph disconnected: cannot obtain value for tensor KerasTensor
TensorFlow版本:2.4.0
最近在嚐試寫PGGAN,參考來源程式碼在過程中做了轉型:
input_img = tf.keras.layers.Input(shape=(4, 4, 3))
input_img = tf.cast(input_img, tf.float32)
x1 = tf.keras.layers.AveragePooling2D()(input_img)
這種情況下每次都會跳出異常訊息:
Graph disconnected: cannot obtain value for tensor KerasTensor
搞了三天三夜,不管怎麼改變寫法,改成lambda
、或是tf.dtypes.cast
都不行:
input_img = tf.dtypes.cast(input_img, tf.float32)
結果我就很生氣的把變數名稱改了…
input_img = tf.keras.layers.Input(shape=(4, 4, 3))
input_img2 = tf.cast(input_img, tf.float32)
x1 = tf.keras.layers.AveragePooling2D()(input_img2)
就好了,對,沒錯,就好了。恭喜我自己,怪了,那他是怎麼訓練完成的。