hswish

小白入门深度学习 | 3-2：激活函数activation

文章目录一、什么是激活函数二、激活函数的种类1.sigmoid2.Tanh3.ReLU4.LReLU5.PReLU6.RReLU7.ELU8.Swish9.hswish三、激活函数的选择四、激活函数相关问题一
K同学啊·2023-09-19 23:15

使用Python API实现TRT版BN/hswish/Silu等算子

文章目录BatchNorm2d的TRT实现BatchNorm定义TRT的Scale层定义Scale实现BNFusedBatchNormalizationBatchNorm1d的TRT实现TRT实现hswishTRT实现Silu算子方法1：转ONNX时替换源码silu算子方法2：TRTAPI间接实现SiluBN与track_running_stats参考BatchNorm2d的TRT实现Tensor
wq_0708·2022-12-14 20:39
上一页 1 下一页
按字母分类： ABCDEFGHIJKLMNOPQRSTUVWXYZ其他