【TensorRT】关于custom plugin的实际code problems汇总

技术博客与活动 历史活动 GPU Hackathon
1

1.在实际定义CustomPlugin::CustomPlugin(some params);和CustomPlugin::CustomPlugin(const nvinfer1::Weights *weights, int nbWeights);如下：
[attach]15988[/attach]
问：什么时候使用这种带Weights的插件构造函数？什么时候使用自己定义的带形参构造函数？区别是什么

2.在重写serialize方法时，需要序列化哪些参数，怎么根据自己需要写的插件来确定需要序列化什么参数？如下
[attach]15987[/attach]
问：自定义插件时如何确定需要序列化的参数，如图中的mClassCount、mThreadCount、mKernelCount、mYoloKernel x N，也就是序列化的是保存什么变量到char*中

3.在createPlugin函数调用时，传入的第二个参数是PluginFieldCollection 类型，是需要传入什么数据？如下图：
[attach]15986[/attach]