firstconv是模型量化的必须步骤吗,哪些模型量化后模型会好一些,大家谁知道
可以不用firstconv吗?
1、firstconv可以把mean和std的运算放在MLU上,减少Host的计算压力,释放CPU资源。2、firstconv可以直接输入UINT8的图片数据,减少了H2D的拷贝数据量。当然这2点是相辅相成的,没有1也无法实现2,因为图片数据一但经过了mean和std运算,UINT8就表达不了了,必须用Float了
1、firstconv可以把mean和std的运算放在MLU上,减少Host的计算压力,释放CPU资源。2、firstconv可以直接输入UINT8的图片数据,减少了H2D的拷贝数据量。
当然这2点是相辅相成的,没有1也无法实现2,因为图片数据一但经过了mean和std运算,UINT8就表达不了了,必须用Float了
firstconv是寒武纪性能优化的一种方式,具体为当模型第一层为conv层时,可将前处理的mean和std的处理放在模型内,以此提升端到端的性能。
请登录后评论