The seq2seq-layout-analysis from limengyang1992

anyconfig.globals.UnknownFileTypeError: No parser found for file 'file extension=yaml'

python test.py

File "test.py", line 7, in
import dataset
File "/seq2seq-ocr-analysis-main/dataset.py", line 9, in
config = anyconfig.load(open("config.yaml", 'rb'))
...
raise UnknownFileTypeError("file extension={}".format(fileext))
anyconfig.globals.UnknownFileTypeError: No parser found for file 'file extension=yaml'

转ONNX问题

您好，我这边将Seq2Seq模型转ONNX模型，然后使用onnxruntime推理计算，输出结果全部为0.，请问这个Seq2Seq模型转ONNX，是需要注意什么吗？

line=list(model[each]) TypeError: 'Word2Vec' object is not subscriptable

实验结果

请问有没有相关的实验结果参考么？

Expected tensor for argument #1 'indices' to have scalar type Long

File "", line 1, in
text_hidden = processing_text(text_list[:max_len])

File "", line 103, in processing_text
out = rnn_infer.predict(label_list)

File "C:\Users\shiqiang42\Desktop\seq2seq-layout-analysis-main\seq2seq-layout-analysis-main\rnn_infer.py", line 46, in predict
types, hidden = model(input_tensor)

File "D:\software\anaconda3\envs\ocr\lib\site-packages\torch\nn\modules\module.py", line 550, in call
result = self.forward(*input, **kwargs)

File "C:\Users\shiqiang42\Desktop\seq2seq-layout-analysis-main\seq2seq-layout-analysis-main\models\TextRNN.py", line 44, in forward
emb = self.embedding(x)

File "D:\software\anaconda3\envs\ocr\lib\site-packages\torch\nn\modules\module.py", line 550, in call
result = self.forward(*input, **kwargs)

File "D:\software\anaconda3\envs\ocr\lib\site-packages\torch\nn\modules\sparse.py", line 114, in forward
self.norm_type, self.scale_grad_by_freq, self.sparse)

File "D:\software\anaconda3\envs\ocr\lib\site-packages\torch\nn\functional.py", line 1724, in embedding
return torch.embedding(weight, input, padding_idx, scale_grad_by_freq, sparse)

RuntimeError: Expected tensor for argument #1 'indices' to have scalar type Long; but got torch.IntTensor instead (while checking arguments for embedding)

text_hidden = processing_text(int(text_list[:max_len]))
Traceback (most recent call last):

File "", line 1, in
text_hidden = processing_text(int(text_list[:max_len]))

TypeError: int() argument must be a string, a bytes-like object or a number, not 'list'

text_hidden = processing_text(text_list[:max_len])
Traceback (most recent call last):