2020년 4월 13일 월요일

tensorflow 2.1 - keras.utils.Sequence workers >1 when finalizing GeneratorDataset iterator

tensorflow 2.1 에서 keras.utils.Sequence  사용시  workers를 1이상 사용하면, 1epochs 끝날때 마다 when finalizing GeneratorDataset iterator warning을 내 뱉는다.  그대로 정상 수행 되는거 같지만, cpu 메모리 가 점점늘어난다.

문제점은 찾은 상태고, 버그 수정 후 병합 대기 중이니, 지금 2.2 까지 나온 상태이니.. 2.3쯤 되면 정상 사용할 수 있을듯하다..

그전까진 keras.utils.Sequence 를 최대한 자제 (급하면 workers=1로) 하고. tf.data를 사용해야한다.

댓글 없음:

댓글 쓰기