Функция CudnnConvolutionBackward вернула недопустимый градиент с индексом 1 — got [294, 592, 1, 1], но ожидаемая форма, совместимая с [294, 768, 1, 1]

#pytorch #пыторч Вопрос: Я работаю над проектом pytorch (torch = 1.2.0) с учетом сокращения CNN. Я удалил несколько фильтров из модели и попытался переобучить уменьшенную модель. Однако мой код выходит…

Продолжить чтениеФункция CudnnConvolutionBackward вернула недопустимый градиент с индексом 1 — got [294, 592, 1, 1], но ожидаемая форма, совместимая с [294, 768, 1, 1]