论坛
BBS
会员中心
Space
用户
登录
用户名
Email
自动登录
找回密码
密码
登录
立即注册
搜索
搜索
本版
帖子
用户
本版
帖子
用户
CUDA
技术
工具
资源
入门
社区
论坛
开启辅助访问
切换到宽版
好友
帖子
收藏
道具
勋章
任务
设置
我的收藏
退出
腾讯QQ
微信登录
论坛
›
社区信息
›
GPU Hackathon
›
TRT Hackathon
›
查看内容
发表新帖
查看:
7878
|
回复:
0
问:关于informer,现在的informer在输入前有一个embedding层,...
GPUS-Cat
GPUS-Cat
当前离线
GPUS-Cat
管理员
|
显示全部楼层
|
阅读模式
1
#
电梯直达
发表于 2021-4-19 09:59:58
7878
0
问:关于informer,现在的informer在输入前有一个embedding层,用来把输入转换到固定的大小,这个一起转为onnx比较合适呢,还是手写c++实现呢,这个embedding算是一个可以和模型剥离的东西
答复:我觉得如果用 onnx 的方案,就一起转比较好。因为如果一部分用 C++或者 python 去搭 API,另一部分用 onnx,这样的话拼合起来将会很麻烦,可能得用比较 hack 的方式。这个在 torch 导出的时候,是可以指定 dynamic 的维度的。
使用道具
举报
回复
高级模式
B
Color
Image
Link
Quote
Code
Smilies
您需要登录后才可以回帖
登录
|
立即注册
发表回复