用户
 找回密码
 立即注册
发表于 2021-4-19 09:59:58
78780
问:关于informer,现在的informer在输入前有一个embedding层,用来把输入转换到固定的大小,这个一起转为onnx比较合适呢,还是手写c++实现呢,这个embedding算是一个可以和模型剥离的东西
答复:我觉得如果用 onnx 的方案,就一起转比较好。因为如果一部分用 C++或者 python 去搭 API,另一部分用 onnx,这样的话拼合起来将会很麻烦,可能得用比较 hack 的方式。这个在 torch 导出的时候,是可以指定 dynamic 的维度的。

使用道具 举报 回复
发新帖
您需要登录后才可以回帖 登录 | 立即注册