logo
Loading...

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

因為LSTM訓練較慢,所以嘗試使用colab 的TPU做訓練使用tensorflow 2.x以及te...

tpu

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

2020/06/14 09:45 AM
機器學習新手論壇
Eliana
觀看數:0
回答數:1
收藏數:0
tpu

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

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


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

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

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


但是使用GPU是沒有問題的

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

所以來問問看

謝謝!