你好,开发者。
当我将layoutxlm RE模型导出到ONNX时,我发现torch.nn.bilinear无法导出到ONNX。是否有一个可以被ONNX支持的函数来替换torch.nn.bilinear?
kninwzqo1#
你好,双线性层可以重写为一个新的类。我还有一个问题,当将layoutlmv2 RE模型转换为onnx时,输入的'entities'和'relations'是字典类型,如何进行转换?
k75qkfdt2#
@gjj123 Could you help me know how to write bilinear to a new class
nxowjjhe3#
@syannghttps://blog.csdn.net/gjj19901005/article/details/127530571?ops_request_misc=%257B%2522request%255Fid%2522%253A%2522169408710616800188532010%2522%252C%2522scm%2522%253A%252220140713.130102334..%2522%257D&request_id=169408710616800188532010&biz_id=0&utm_medium=distribute.pc_search_result.none-task-blog-2~all~baidu_landing_v2~default-1-127530571-null-null.142^v93^chatgptT3_2&utm_term=layoutlmv2%20onnx&spm=1018.2226.3001.4187
3条答案
按热度按时间kninwzqo1#
你好,双线性层可以重写为一个新的类。我还有一个问题,当将layoutlmv2 RE模型转换为onnx时,输入的'entities'和'relations'是字典类型,如何进行转换?
k75qkfdt2#
@gjj123 Could you help me know how to write bilinear to a new class
nxowjjhe3#
@syanng
https://blog.csdn.net/gjj19901005/article/details/127530571?ops_request_misc=%257B%2522request%255Fid%2522%253A%2522169408710616800188532010%2522%252C%2522scm%2522%253A%252220140713.130102334..%2522%257D&request_id=169408710616800188532010&biz_id=0&utm_medium=distribute.pc_search_result.none-task-blog-2~all~baidu_landing_v2~default-1-127530571-null-null.142^v93^chatgptT3_2&utm_term=layoutlmv2%20onnx&spm=1018.2226.3001.4187