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를 사용해야한다.
댓글 없음:
댓글 쓰기