[에러 해결 방법] WARNING:tensorflow:Your input ran out of data

이 글은 2021년 당시 TensorFlow 2.4.0 기준으로 작성된 설치/사용 기록이다. 최신 버전에서는 명령이 다를 수 있다.

TensorFlow 훈련 중 ‘Your input ran out of data’ WARNING이 발생하는 문제를 다룬다. 원인은 TensorFlow 버전 문제이며, 2.2.0으로 다운그레이드하여 해결한다.

에러 메시지

WARNING:tensorflow:Your input ran out of data; interrupting training. Make sure that your dataset or generator can generate at least 'steps_per_epoch * epochs' batches

원인

• TENSORFLOW 버전 문제 (본인 기준 Tensorflow=2.4.0)

해결 방법

1. tensorflow를 2.2.0으로 다운 그레이드 해주면 된다.

참고

• tensorflow, keras 오류 해결하기