重载IInt8EntropyCalibrator2中getBatch函数时出错
zhjw0927 opened this issue · 0 comments
zhjw0927 commented
hi,我在量化模型时,继承了IInt8EntropyCalibrator2类。在继承类的构造函数中,初始化了sizaMap<std::string, size_t> 成员变量,方便在getBatch函数中去调取不同input node的size。但是,我发现在量化过程运行时(即调用getBatch时),使用input node name去选取sizeMap时,发生了错误。debug发现,构造函数中初始化的sizeMap在成员函数getBatch中,key发生了乱码。这样的问题,也发生在readCalibrationCache中。