rlcode / reinforcement-learning-kr-v2

[파이썬과 케라스로 배우는 강화학습] 텐서플로우 2.0 개정판 예제
MIT License
128 stars 100 forks source link

텐서플로 2.x 속도 이슈는 없나요? #17

Open rubyjohn opened 3 years ago

rubyjohn commented 3 years ago

1판의 내용을 텐서플로 2.0 나오고 텐서플로.org 사이트의 가이드에 따라 억지로 고쳐서 했었는데 속도가 너무 느려서 포기했더랬습니다.

그런데 이곳의 2판 샘플은 제가 고친내용이랑 꽤나 다르군요.

그래서 저렇게 쓰면 뭔가 좀 개선이 있나 싶어 직접 비교하려고 보니 모델을 고친 부분도 있고 여러가지 고친 사항이 많더군요. 완전히 똑같이 맞춰서 고치려다 보니 tf.train의 Rmsprop에는 rho도 없고 똑같이하는게 어렵더군요. 또 괜히 시간만 잡아먹는거 아닌가 싶어서 저자님의 의견을 듣고 싶습니다.

TF 2.x의 속도 문제는 제대로 개선된건가요?