[Model Train Error] 모델 학습 시 무한 루프 /Unknown batch

3498/Unknown - 10ms/step - loss: 0.1356 -    

3585/Unknown - 1029s 282ms/step - loss: 0.13   

3586/Unknown - 1029s 282ms/step - loss: 0.1323 - accuracy: 0.   

3587/1030s 282ms/step - loss: 0.1322 - accuracy: 0.   

 

ImageGenerator를 사용하고 model을 학습 시키면 generator가 image의 개수를 알지 못해 무한이 학습을 반복한다. (28시간 학습.. )

 

 

https://stackoverflow.com/questions/61266275/epoch-1-2-103-unknown-8s-80ms-step-loss-0-0175-model-fit-keeps-running-f

Epoch 1/2 103/Unknown - 8s 80ms/step - loss: 0.0175 (model.fit() keeps running forever even after crossing the total number of t

# Solution

model 학습 부분에 steps_per_epoch = 학습 이미지 수를 추가해주면 된다.