New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

pytorch전용 v3.0 GPU 질문 #130

Open

ChulgooKim opened this issue Jan 4, 2022 · 1 comment

Labels

v3

ChulgooKim commented Jan 4, 2022

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

Owner

quantylab commented Jan 5, 2022

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

quantylab added the v3 label

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment