在网络模型910推理迁移完毕之后,我开始根据ModelZoo库中的代码编写310推理代码,但是编写310推理的时候出现了一个out of memory错误。我使用的模型总共使用了6层dprnn block,使用5层block可以训练出ckpt文件并export出mindir文件,310推理可以正常运行。而当使用6层block的时候,可以训练出ckpt文件,也可以export出mindir文件,但是310推理报out of memory错误,具体报错信息如下:
![](https://i-blog.csdnimg.cn/blog_migrate/d4eeb2e1c5ba76b45e5a10040282bf0e.png)
检查了export代码以及310代码,并没有可以导致爆内存的操作。export部分的主代码我粘贴在了下面,供大家参考。