rltrader icon indicating copy to clipboard operation
rltrader copied to clipboard

Published 20 hours ago • verified publisher icon quantylab

pytorch전용 v3.0 GPU 질문

Open ChulgooKim opened this issue 3 years ago • 1 comments

v3.0 에서 cuda를 사용하셨는데 predict 함수에서 device간에 이동이 많이 이뤄져서 속도가 저하되는것 같습니다. 혹시 개선중에 있는지 궁금합니다. 개인적으로도 개선해 보려고도 하고있습니다.

ChulgooKim avatar Jan 04 '22 03:01 ChulgooKim

@ChulgooKim predict 수를 줄일 수 있다면 좋겠지만 강화학습 특성상 좀 어려운 부분이 있네요... 혹시 좋은 개선안을 찾으시면 공유해주시면 감사하겠습니다!

quantylab avatar Jan 05 '22 04:01 quantylab