pytorch dataloader가 epoch 한번 돈 후에 멈춰버리는 현상

yeonsoo·2022년 5월 2일
0

현상: 학습 시 pytorch dataloader가 1 epoch 이후로 멈춰버림. 중단지점은 next_data()의 timeout.

도저히 해결을 못하고 있다가 이해할 수 없는 해결책을 찾음.
import torch 후에 import cv2가 오도록 import 순서를 바꿔주면 문제 없이 돌아간다.
출처: https://github.com/pytorch/pytorch/issues/33296

profile
to be enterprising

0개의 댓글