pytorch: при отладке цикла dataLoader for он сильно зависает и не может войти в точку останова в преобразовании.

Рекомендуемое решение:

  • Прежде чем предыдущий процесс завершит обработку, он переходит к следующему, вызывая блокировку.После завершения эпохи или после получения данных каждого пакета наступает пауза: time.sleep(0.003)
  • Проблемы с памятью, используйте переключатель: pin_memory=True/False
  • Несколько процессов вызывают проблемы блокировки, уменьшите количество процессов или используйте один напрямую: num_workers=0/1
  • При проблемах с использованием других DataLoader измените: from torch.utils.data.dataloader import DataLoader
  • Проблема недостаточного объема памяти: запись 8192 в /proc/sys/kernel/shmmni
  • Если OpenCV используется в сценарии одновременно, это может быть проблемой блокировки между OpenCV и Pytorch.
  • Отключите многопоточность OpenCV:
  • cv2.setNumThreads(0)
  • cv2.ocl.setUseOpenCL (ложь)
  • Есть ли проблема с тем, что открытый файл не закрывается: [openfile].close()

Guess you like

Origin blog.csdn.net/m0_72734364/article/details/133427829