pytorch_triple_extraction 训练好的模型在测试时发生了维度错误。

训练好的模型在测试时发生了维度错误。

Open Lv-shuai opened this issue 2 years ago • 7 comments

屏幕截图 2022-10-15 010322 chinese-bert-wwm-ext与chinese-roberta-wwm-ext里面的文件是一样的吗？我看他们的大小都相同。

Oct 14 '22 17:10 Lv-shuai

我将train batch和test batch都改成了1，是这个原因吗

Oct 14 '22 17:10 Lv-shuai

chinese-bert-wwm-ext与chinese-roberta-wwm-ext里面的文件是一样的吗？我看他们的大小都相同。

应该是一样的。

Oct 15 '22 00:10 taishan1994

我将train batch和test batch都改成了1，是这个原因吗

要先确保数据的长度都在设置的最大长度内。

Oct 15 '22 00:10 taishan1994

数据是题目里的数据，进过process.py操作，分割成的train和eval。数据的长度应该都在最大长度512以内吧。

Oct 15 '22 01:10 Lv-shuai

数据是题目里的数据，进过process.py操作，分割成的train和eval。数据的长度应该都在最大长度512以内吧。

你可以打印下报错的那个地方的两个数据的维度看看先。

Oct 15 '22 01:10 taishan1994

您的numpy版本是多少？下面这个问题我感觉是numpy版本原因。屏幕截图 2022-10-15 234850

Oct 15 '22 15:10 Lv-shuai

您的numpy版本是多少？下面这个问题我感觉是numpy版本原因。

1.21.6

Oct 16 '22 01:10 taishan1994

chinese-bert-wwm-ext与chinese-roberta-wwm-ext里面的文件是一样的吗？我看他们的大小都相同。

您好，请问您解决这个问题了吗？我遇到了相同的问题。

Dec 31 '22 10:12 shunmine00

chinese-bert-wwm-ext与chinese-roberta-wwm-ext里面的文件是一样的吗？我看他们的大小都相同。

您好，请问您解决这个问题了吗？我遇到了相同的问题。

你重新设置一下那两个权重文件夹，或者看看你的环境。可能是环境不一样，有的包版本问题。

Jan 06 '23 15:01 Lv-shuai