ConcatOp: Dimensions of inputs should match shape ... error가 나올 때

에러

 

만약 에러가 train되는 마지막에 등장했다면 이 해법은 확실하다.

 

batch를 활용해서 train하고 있을 때, 아마 마지막에 batch_size가 동일 하지 않기 때문에 발생된 에러이다.

예를 들어

dataset = Dataset().batch(batch_size)

이런 식으로 되어 있다면 이거를 

dataset = Dataset().batch(batch_size, drop_remainder = True)

뒤에 단어를 추가해주면 간단하게 문제가 fix 된다. 

 

drop_remainder를 True로 설정하면 마지막 배치 크기를 무시하고 지정한 배치크기를 유지할 수 있다. 그래서 dimension과 관련된 에러없이 문제가 해결된다.