使用 colab TPU (tf.keras) 無法預測正確數量

2020/06/14 下午 05:45

機器學習共學討論版

Eliana

觀看數：6

回答數：1

收藏數：0

tpu

因為LSTM訓練較慢，所以嘗試使用colab 的TPU做訓練

使用tensorflow 2.x以及tensorflow底下的Keras做訓練與預測

有正確使用到TPU(因為明顯快很多)

但是使用model.predict()的話，預測資料數量都是16的倍數(batch size設128)

即使只預測5筆，但是predict出來還是16筆

但是使用GPU是沒有問題的

網路上也沒有查到有遇到這樣問題的人

所以來問問看

謝謝!