实现多任务并行的深度网络训练方法和系统
基本信息
申请号 | CN202110518804.0 | 申请日 | - |
公开(公告)号 | CN113435571A | 公开(公告)日 | 2021-09-24 |
申请公布号 | CN113435571A | 申请公布日 | 2021-09-24 |
分类号 | G06N3/04(2006.01)I;G06K9/62(2006.01)I | 分类 | 计算;推算;计数; |
发明人 | 叶杜楠;郭骏;潘正颐;侯大为;闫科萍 | 申请(专利权)人 | 上海微亿智造科技有限公司 |
代理机构 | 上海段和段律师事务所 | 代理人 | 李佳俊;郭国中 |
地址 | 201100上海市闵行区元江路5500号第1幢 | ||
法律状态 | - |
摘要
摘要 | 本发明提供了一种实现多任务并行的深度网络训练方法和系统,包括:步骤1:设置卷积网络,将多个相关联的子任务连接并共享该卷积网络;步骤2:设置学习率和迭代次数,交替训练不同的网络分支;步骤3:在网络分支的训练效果达到预设标准后停止训练,并输出该网络分支头部代表的任务。本发明将多个任务一起学习,多个任务共享一个模型可以减小模型内存的占用,同时多个任务可以通过一次向前计算得到结果,加快了推理速度。 |
