본문 바로가기
Python_Tensorflow

[Python_MachineLearning] TFRecord사용하면서 겪은 이슈

by 신입같은 3년차 2018. 6. 28.

TFRecord파일을 Trainning 시키고 Predict까지 TestDataSet으로 돌리면서 느리다는 생각을 해본적이 없었습니다.


하지만 label이 없는 X 데이터 하나만 넣었을때 0.27 초가량이 소요되었고 10개를 확인하는데 3초정도가 소요되는것을 확인했습니다.


딱봐도 이상하다해서 확인해본결과 Estimator를 사용하여 predict를 할때 하나씩 넣는게 아닌 2차원 numpy.ndarray타입으로 한번에 집어넣었고 1개가 되었든 100개가 되었든 얼추 비슷한 시간이 소요되는것을 확인하였습니다.


이처럼 Estimator를 사용하여 여러개의 인식을 할때는 한번에 넣어주는걸 권장합니다 ...


초보의 관점에서 발생한 이슈를 정리해보았습니다.

반응형

댓글