大小不匹配错误
RuntimeError: 加载 NougatModel 的状态字典时出现错误: encoder.model.layers.1.downsample.norm.weight 的大小不匹配: 从检查点中拷贝形状为 torch.Size([1024]) 的参数,当前模型中的形状为 torch.Size([512])。 encoder.model.layers.1.downsample.norm.bias 的大小不匹配: 从检查点中拷贝形状为 torch.Size([1024]) 的参数,当前模型中的形状为 torch.Size([512])。 encoder.model.layers.1.downsample.reduction.weight 的大小不匹配: 从检查点中拷贝形状为 torch.Size([512, 1024]) 的参数,当前模型中的形状为 torch.Size([256, 512])。 encoder.model.layers.2.downsample.norm.weight 的大小不匹配: 从检查点中拷贝形状为 torch.Size([2048]) 的参数,当前模型中的形状为 torch.Size([1024])。 encoder.model.layers.2.downsample.norm.bias 的大小不匹配: 从检查点中拷贝形状为 torch.Size([2048]) 的参数,当前模型中的形状为 torch.Size([1024])。 encoder.model.layers.2.downsample.reduction.weight 的大小不匹配: 从检查点中拷贝形状为 torch.Size([1024, 2048]) 的参数,当前模型中的形状为 torch.Size([512, 1024])。 您可以考虑在模型的 `from_pretrained` 方法中添加 `ignore_mismatched_sizes=True`。
内容来源: facebookresearch/nougat