摘要
本发明公开了一种基于迁移学习的AI深度学习模型训练方法及装置。其中,该方法包括:选择预训练模型,通过移除所述预训练模型的最后一层分类器得到截断模型;在所述截断模型上,添加一浅层全连接分类器,得到浏览器端深度学习模型;利用训练数据训练所述浏览器端深度学习模型的所述浅层全连接分类器。本发明解决了浏览器中深度学习模型从头开始训练复杂且资源消耗大的问题的技术问题。
技术关键词
预训练模型
分类器
深度学习模型训练
模型拓扑结构
可读存储介质
数据
计算机程序产品
处理器
存储器
计算机设备
优化器
编程
参数
模块
图片
标签
标记