3

我正在尝试在 AWS EC2 实例上使用 fastai 启动一个 Web 应用程序,但在检查点和当前模型之间出现大小不匹配错误。

但是,这个网络应用程序在我的本地机器上使用相同的文件运行良好,我已经在本地测试了几天,它仍然没有遇到这个问题。

有什么建议吗?我认为这与软件包版本有关,但到目前为止我还没有得到任何结果。

RuntimeError: Error(s) in loading state_dict for AWD_LSTM:
    size mismatch for encoder.weight: copying a param with shape torch.Size([6399, 400]) from checkpoint, the shape in current model is torch.Size([6389, 400]).
    size mismatch for encoder_dp.emb.weight: copying a param with shape torch.Size([6399, 400]) from checkpoint, the shape in current model is torch.Size([6389, 400]).
4

0 回答 0