[디버깅] WARNING:tensorflow:Your input ran out of data; interrupting training.

케라스 창시자에게 배우는 딥러닝 예시 코드 5-14 실행 중 다음과 같은 오류 발생

Found 2000 images belonging to 2 classes.

Found 1000 images belonging to 2 classes.

Epoch 1/100

 63/100 [=================>............] - ETA: 9s - loss: 0.6940 - acc: 0.5095WARNING: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 (in this case, 10000 batches). You may need to use the repeat() function when building your dataset.

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 (in this case, 50 batches). You may need to use the repeat() function when building your dataset.

 63/100 [=================>............] - 20s 316ms/step - loss: 0.6940 - acc: 0.5095 - val_loss: 0.6906 - val_acc: 0.5000

 

 

구글링 결과 역자이신 박해선님의 블로그에서 keras 버전에 대한 정보 입수.

keras는 2.3.1, tensorflow는 2.2.0 로 재설치했다.

---

과정

1. 삭제하기

# 기존 keras 삭제하기
!pip uninstall keras   # colab에서 하는 경우
!pip uninstall tensorflow

pip uninstall keras    # 로컬에서 하는 경우
pip install tensorflow

하고 설치하기 전에 런타임을 다시 시작해야 함.

왜냐하면 삭제하기 전에 이미 keras를 import 했기 때문. 그냥 그 상태에서 다시 설치하려고 하면 오류 남.

---

2. 재설치하기

# 코랩에서 하는 경우
!pip install keras==2.3.1
!pip install tensorflow==2.2.0

# 로컬에서 하는 경우
pip install keras==2.3.1
pip install tensorflow==2.2.0

라고 치면 keras가 정상적으로 재설치 되고, 코드도 오류 없이 돌아간다.

부록(tensorflow, keras 버전 확인하기)

# 텐서플로우 버전 확인하기 (굳이 as tf를 하지 않고 tensorflow.__version__을 해도 됨)
import tensorflow as tf
print(tf.__version__)


# 케라스 버전 확인하기
import keras
print(keras.__version__)