Merge pull request #3115 from diandianti/master

MNN:BugFix:fix bug when use quantized.out with multi-images
alibaba · Dec 6, 2024 · 2b899c1 · 2b899c1
2 parents 98ba45b + 53d45b5
commit 2b899c1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tools/quantization/calibration.cpp b/tools/quantization/calibration.cpp
@@ -1056,7 +1056,7 @@ void Calibration::_quantizeModelEMA() {
                 auto singleInput = _Input(originDims, originFormat, originType);
                 auto inputTensor = (MNN::Tensor*)singleInput->getTensor();
                 Helper::preprocessInput(_process.get(), _preprocessConfig, file, inputTensor, _inputType, inputs[0]);
-                ::memcpy(inputs[0]->writeMap<float>() + k * inputTensor->elementSize(), inputTensor->host<float>(), inputTensor->elementSize() * sizeof(float));
+                ::memcpy(inputs[0]->writeMap<float>() + (k - indicesStart) * inputTensor->elementSize(), inputTensor->host<float>(), inputTensor->elementSize() * sizeof(float));
 
             }
         }