标签: pytorch transfer
我正在使用预训练权重实施迁移学习。我只做特征提取,而不是对整个模型进行微调。我已经在 ResNet50 预训练模型上初始化了线性层。 我有视频输入。 我收到以下运行时错误: RuntimeError: 4 维权重 [64, 3, 7, 7] 的预期 4 维输入,但得到大小为 [30, 60, 3, 224, 224] 的 5 维输入。
(批量大小为 30,样本持续时间(输入的时间持续时间)为 60)
谁能帮我解决这个问题?