下载一种模型训练显存优化方法及装置的技术资料

文档序号:43844378

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

本发明公开了一种模型训练显存优化方法及装置,其中,优化方法包括:基于预设网络模型,训练一个轮次,记录每个张量的输入时刻、输出时刻、显存占用量和计算时间;根据传输速度和显存占用量,计算每个张量从GPU显存转移到主内存的转移时间开销;根据每个张...
该专利属于浙江大学所有,仅供学习研究参考,未经过浙江大学授权不得商用。

详细技术文档下载地址

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