Передача обучения в Pytorch: проблема измерения ввода

#pytorch #transfer

Вопрос:

Я внедряю трансфертное обучение с использованием предварительно подготовленных весов. Я занимаюсь только извлечением функций, а не тонкой настройкой всей модели. Я инициализировал линейный слой на предварительно обученной модели ResNet50. У меня есть видеовходы . Я получаю следующую ошибку во время выполнения: Ошибка времени выполнения: Ожидаемый 4-мерный ввод для 4-мерного веса [64, 3, 7, 7], но получил 5-мерный ввод размера [30, 60, 3, 224, 224] вместо этого.

(размер пакета-30, длительность выборки(временная длительность входных данных) — 60)

Может ли кто-нибудь, пожалуйста, помочь мне решить эту проблему?