一种ViT模型的硬件加速方法、装置、设备及存储介质

本发明公开了一种ViT模型的硬件加速方法、装置、设备及存储介质，设计计算机视觉技术领域，包括：利用批量归一化层及预设目标函数分别对初始ViT模型中的线性层与非线性层进行调整，得到调整后ViT模型；预设目标函数包括线性整流函数和以2为底的目标归一化函数；基于预设数据集及预设模型训练规则对调整后ViT模型进行模型训练，得到目标ViT模型；分别为目标ViT模型的线性层和非线性层设计加速电路，得到加速电路设计结果；基于加速电路设计结果进行全流水架构硬件加速器的构建，并利用得到的目标全流水硬件加速器对目标ViT模型进行加速。能够在兼顾考虑线性层与非线性层的改变的条件下，减小ViT模型规模，实现硬件加速。