활연개랑

[tensorflow] error해결 : train_function -> train_function -> train_function -> train_function 오류 본문

딥러닝

[tensorflow] error해결 : train_function -> train_function -> train_function -> train_function 오류

승해tmdhey 2022. 6. 10. 15:53
반응형

 

Function call stack:
train_function -> train_function -> train_function -> train_function

다음 오류는 network에 들어가는 input_shape과, 실제 데이터의 input_shape이 다른 경우 나올 수 있습니다.

shape이 정확하게 맞도록 설정해주세요.

 

또한 tfrecord같은 dataset에서 parsing 하는 과정에, type이 제대로 바뀌지 않을 경우에도 발생할 수 있으니 parsing함수를 다시 한 번 확인해보시면 되겠습니다. 

 

보통 이 오류는 들어가는 data에 대한 문제이니, 위 두 가지를 확인해 보셨는데도 해결되지 않는다면  input data를 가져오는 전체적인 과정을 쭉 살펴보며 디버깅 해보시면 문제를 해결하실 수 있을 것입니다. 굿