[오류 해결] AttributeError: 'Mean' object has no attribute 'reset_states'.

발생한 오류

atari 환경에서 강화학습 훈련을 시키려고 할 때 다음과 같은 오류가 발생하였다. 

AttributeError: 'Mean' object has no attribute 'reset_states'.

 

reset_state 속성이 없다는 의미이다. 

 

 

해결 방법

특정 버전의 tensorflow에서 reset_state를 지원하지 않아서 생기는 문제였다.

TF 2.4.1의 경우 m.reset_states(), TF 2.5.0에는 m.reset_state()를 사용해야 한다고 한다!

 

버전에 맞게 수정하면 잘 동작한다. 

reset_state()
reset_states()

 

 

Reference

https://github.com/tensorflow/tensorflow/issues/50359