void* mkldnn::impl::memory_tracking::registry_t::get(const key_t&, void*) const: Assertion `size() == 0' failed.

问题来源:用CPU进行模型测试(测试某张图片时出现问题,改图片之前的测试是正常的)

问题环境:
python  3.6.2 

torch   1.3.1 

CPU信息

 

 

解决办法:换用大显存的GPU测试

目前还没有其他好的解决办法,希望路过的大佬能提供帮助

posted on 2021-04-21 15:05  cltt  阅读(136)  评论(0编辑  收藏  举报

导航