首页 > 解决方案 > 如何解决在pytorch中加载预训练模型时的大小不匹配问题?

问题描述

我正在尝试将卷积层的过滤器(通道)数量为 的预训练模型加载到卷积层中64具有128过滤器的模型中。我使用new_network.load_state_dict(pretrained_model, strict=False)但我面临所有这些层的大小不匹配错误。如何使用可用的预训练模型权重初始化每个卷积层中的一些过滤器?

标签: pythondeep-learningpytorchconv-neural-networktransfer-learning

解决方案


推荐阅读