RuntimeError: shape ‘[2097152, 1]‘ is invalid for input of size 1048576
于是查看模型的输入，model.shape是1024，然后再看数据的类型.shape后是（731，2048）。数据的维度是2048，而模型的输入数据是1024，导致的错误。接受输入形状为[2097152,1]，但是数据输入是1048567。可以发现两个数字是倍数关系。这里有两个方法，该数据维度或者模型输入维度。于是了解probs = model(seq)，是model函数或模型的输出。报错显示该错
·
接受输入形状为[2097152,1]，但是数据输入是1048567。可以发现两个数字是倍数关系
在一篇博客看到1048567=128*128*64
报错显示该错误与probs = model(seq) 相关
于是了解probs = model(seq)，是model函数或模型的输出。
于是查看模型的输入，model.shape是1024，然后再看数据的类型.shape后是（731，2048）。数据的维度是2048，而模型的输入数据是1024，导致的错误。
这里有两个方法，该数据维度或者模型输入维度。我选择了修改模型维度，于是成功解决问题
魔乐社区（Modelers.cn) 是一个中立、公益的人工智能社区，提供人工智能工具、模型、数据的托管、展示与应用协同服务，为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作，由全产业链共同建设、共同运营、共同享有，推动国产AI生态繁荣发展。
更多推荐
3
0- 0
已为社区贡献2条内容
所有评论(0)