diff --git a/cpp/ptq/main.cpp b/cpp/ptq/main.cpp index 8b73bca33b..c5cc97bcec 100644 --- a/cpp/ptq/main.cpp +++ b/cpp/ptq/main.cpp @@ -13,7 +13,6 @@ #include int main(int argc, const char* argv[]) { - trtorch::logging::set_reportable_log_level(trtorch::logging::Level::kERROR); if (argc < 3) { std::cerr << "usage: ptq \n"; return -1; @@ -42,7 +41,7 @@ int main(int argc, const char* argv[]) { std::string calibration_cache_file = "/tmp/vgg16_TRT_ptq_calibration.cache"; - auto calibrator = trtorch::ptq::make_int8_calibrator(std::move(calibration_dataloader), calibration_cache_file, false); + auto calibrator = trtorch::ptq::make_int8_calibrator(std::move(calibration_dataloader), calibration_cache_file, true); //auto calibrator = trtorch::ptq::make_int8_cache_calibrator(calibration_cache_file);