看板 [ Python ]
討論串[問題] tensorflow- 訓練在gpu 預測在cpu
共 2 篇文章
內容預覽： 開啟 | 關閉 | 只限未讀
首頁
上一頁
1
下一頁
尾頁
#1
[問題] tensorflow- 訓練在gpu 預測在cpu
推噓2(2推 0噓 18→)留言20則，0人參與, 2年前最新作者snakei14702 (阿中是我們的神)時間2年前 (2022/09/22 13:41), 2年前編輯資訊
1
0
0
內容預覽:
最近在玩強化訓練的時候遇到一點小問題就是. 因為模型的深度很深需要GPU來跑. 但是在run env 計算(中間有用到model.predict)一些東西的時候. 如果用GPU跑就顯得太慢沒效率.... 所以想要請教一下各位有沒有辦法在fit model時用GPU. 然後用model.prict的時
(還有845個字)
#2
Re: [問題] tensorflow- 訓練在gpu 預測在cpu
推噓1(1推 0噓 0→)留言1則，0人參與, 2年前最新作者snakei14702 (阿中是我們的神)時間2年前 (2022/09/25 03:11), 編輯資訊
0
0
0
內容預覽:
謝謝各位強者的幫忙. 後來研究一下. 用@tf.fuction修飾. 然後用model(x)取代model.predict(x). 就可以跑的爆炸快了！！！. 以前覺得@tf.fuction好像也沒多厲害. 不過在RL玩遊戲的時候整個讓我很驚艷！. 再次感謝各位的幫忙！. --. ※ 發信站: 批踢
首頁
上一頁
1
下一頁
尾頁