下载一种基于强化学习的算子自动调优方法及相关装置的技术资料

文档序号:39056864

温馨提示:您尚未登录,请点 登陆 后下载,如果您还没有账户请点 注册 ,登陆完成后,请刷新本页查看技术详细信息。

本发明公开了一种基于强化学习的算子自动调优方法及相关装置,基于长短时记忆网络的策略网络,策略网络会顺序地选择各个待调优参数的值,调优参数组合对应的算子实现在目标硬件上的实测算力值将作为奖励信号,使用REINFORCE算法计算梯度来更新策略网...
该专利属于西安交通大学所有,仅供学习研究参考,未经过西安交通大学授权不得商用。

详细技术文档下载地址

温馨提示:您尚未登录,请点 登陆 后下载,如果您还没有账户请点 注册 ,登陆完成后,请刷新本页查看技术详细信息。