标签： python deep-learning pytorch conv-neural-network transfer-learning
我正在尝试将conv层的过滤器（通道）数为64的预训练模型加载到conv层中具有128过滤器的模型中。我使用new_network.load_state_dict(pretrained_model, strict=False)，但所有这些图层都遇到尺寸不匹配错误。 如何使用可用的预训练模型权重初始化每个转换层中的某些过滤器？
64
128
new_network.load_state_dict(pretrained_model, strict=False)