Я пытаюсь запустить модели глубокого обучения, используя pytorch. вот ссылка на модель: https://github.com/JiaRenChang/PSMNet https://github.com/KinglittleQ/PSMNet
Я сталкиваюсь с той же проблемой в этих двух моделях. Я использую 4 Taitan XP GPU, но модель может начать обучение только тогда, когда размер пакета установлен на 1 , когда я устанавливаю размер пакета на другой номер (даже 2), он застрянет до первой эпохи обучения.
мой тестовый отпечаток показывает, что он остановится перед запуском первого слоя свертки.