2020년 4월 5일 일요일

Error occurred when finalizing GeneratorDataset iterator: Cancelled

Error occurred when finalizing GeneratorDataset iterator: Cancelled 라는 에러 가 tensorflow 2.1 업데이트 후 발생하는데, tf.keras.Sequence 에 multi gpu 관련 bug가 있어서 이런 현상 발생하고 있다.
추후 2.2 ver 나오면 해결 될 예정인것 같고, Sequnce를 안쓰고 tf.data.Dataset을 이용해 구현하거나, 임시 방편으로는 worker 를 1로 고정해서 쓴다.

댓글 없음:

댓글 쓰기