使用 colab TPU (tf.keras) 無法預測正確數量
2020/06/14 05:45 下午
機器學習共學討論版
Eliana
觀看數：0
回答數：1
收藏數：0
tpu
因為LSTM訓練較慢，所以嘗試使用colab 的TPU做訓練
使用tensorflow 2.x以及tensorflow底下的Keras做訓練與預測
有正確使用到TPU(因為明顯快很多)
但是使用model.predict()的話，預測資料數量都是16的倍數(batch size設128)
即使只預測5筆，但是predict出來還是16筆
但是使用GPU是沒有問題的
網路上也沒有查到有遇到這樣問題的人
所以來問問看
謝謝!