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测试

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

原文地址:https://www.cnblogs.com/tingtin/p/14685299.html