模型训练时报错提示dev cache overflow
2023/09/14
50
问题信息
问题来源 | 产品大类 | 产品子类 | 关键字 |
---|---|---|---|
官方 | 模型训练 | 应用开发MindX SDK | dev cache overflow |
问题现象描述
使用mxRec进行模型训练时报错并提示“dev cache overflow”。
原因分析
模型中“dev_vocab_size”参数值过小。
解决方案
请修改模型中“dev_vocab_size”参数(device vocabulary size),将该值适当调大。
过大的值可能造成设备侧报out of memory错误,请根据实际训练情况和设备HBM存储大小适度调整。
本页内容