根据工作要求,特征提取更改了模型。同事传了一个TensorRT engine文件。一加载,出错如下:
python3.5: cudnnEngine.cpp:710: bool nvinfer1::cudnn::Engine::deserialize(const void*, std::size_t, nvinfer1::IGpuAllocator&, nvinfer1::IPluginFactory*): Assertion `size >= bsize && "Mismatch between allocated memory size and expected size of serialized engine."' failed.
人们不禁要问……表问了,赶紧想办法解决。
首先,网上搜索了一下,Stackoverflow说有一个类似的错误,没有人回答。这个是一个同事转换的模型,于是问这位高手,把源码、模型都发吾。然后一对比,发现体积相差太大,试了一下就成功了。