rlcode / reinforcement-learning-kr

[파이썬과 케라스로 배우는 강화학습] 예제
MIT License
371 stars 228 forks source link

cartpole-dqn에서 load_model=True시 에러입니다. #20

Open KiJeonNam opened 6 years ago

KiJeonNam commented 6 years ago

안녕하세요. cartpole 예제를 DQN과 A2C로 실행을 하던 중에 error 메세지가 있어서 질문드립니다.

먼저 A2C에서 -카트폴 예제에서의 액터-크리틱(A2C) 에이전트 class A2CAgent: def init(self, state_size, action_size): self.render = True self.load_model = True False들을 True로 수정하여 실행할 경우 학습된 모델을 load해서 실행이 잘되는데요,

DQN에서 경우 -카트폴 예제에서의 DQN 에이전트 class DQNAgent: def init(self, state_size, action_size): self.render = True self.load_model = True 다음과 같이 설정하고 실행할 경우, ValueError: Dimension 1 in both shapes must be equal, but are 24 and 32 for 'Assign_6' (op: 'Assign') with input shapes: [4,24], [4,32]. 와 같은 error창이 생깁니다.

파일들을 새로 받아서 실행해도 똑같은 error가 생겨서요. error가 궁금합니다~!

dnddnjs commented 5 years ago

안녕하세요! 이런 에러라면 아마도 학습시켜놓은 model과 정의되어있는 model의 hidden unit 개수의 차이가 있기 때문에 발생하는 에러같습니다. 혹시 제공하는 코드로 바닥부터 학습시키고 불러와도 에러가 생기나요?