模型性能对比

分类任务

实验设置

属性内容
数据集名称随机数矩阵
平均文本长度30
设备GTX 1080 Ti(11G)
batchsize

258

batchnum100
优化器Adamw

实验结果

模型参数设置参数量推断时间/batch(s)训练时间/batch(s)
Transformer_encoder_dnn
d_model=dim_word, nhead=8,num_head=6
32.13M0.0427s0.2134s
CNN_maxpool_dnn
k_num[512,512,512],
width=[3,4,5],
stride=[1,1,1]
4.64M0.0396 s0.0111 s
GRU_dnn5121.96M0.0117s0.0058s

RNN具有记忆功能,被遍历的单元具有因果联系作用(记忆信息传送),上一时刻隐层的状态参与到了这个时刻的计算过程中,这句话的举例说明就是第一个单元计算的结果会作为第二个单元输入的一部分,所以,当前单元必须等上一单元计算结束,有多少单元就需要计算多少次。

CNN同一层次单元没有因果关系都是等价的,这样就可以依据单元核直接复制出所需所有单元核(参数相同),然后采用矩阵并行运算,只需计算一次。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值