diff --git a/nvdsinfer_custom_impl_Yolo/calibrator.cpp b/nvdsinfer_custom_impl_Yolo/calibrator.cpp
index f5a4408..2cc1e53 100644
--- a/nvdsinfer_custom_impl_Yolo/calibrator.cpp
+++ b/nvdsinfer_custom_impl_Yolo/calibrator.cpp
@@ -30,6 +30,11 @@ namespace nvinfer1
             delete[] batchData;
     }
 
+    int Int8EntropyCalibrator2::getBatchSize() const noexcept
+    {
+        return batchSize;
+    }
+
     bool Int8EntropyCalibrator2::getBatch(void **bindings, const char **names, int nbBindings) noexcept
     {
         if (imageIndex + batchSize > uint(imgPaths.size()))