Skip to content

Commit b989c7f

Browse files
committed
fix(//cpp/ptq): remove some logging from ptq app
Signed-off-by: Naren Dasan <[email protected]> Signed-off-by: Naren Dasan <[email protected]>
1 parent 825be69 commit b989c7f

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

cpp/ptq/main.cpp

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,6 @@
1313
#include <sys/stat.h>
1414

1515
int main(int argc, const char* argv[]) {
16-
trtorch::logging::set_reportable_log_level(trtorch::logging::Level::kERROR);
1716
if (argc < 3) {
1817
std::cerr << "usage: ptq <path-to-module> <path-to-cifar10>\n";
1918
return -1;
@@ -42,7 +41,7 @@ int main(int argc, const char* argv[]) {
4241

4342
std::string calibration_cache_file = "/tmp/vgg16_TRT_ptq_calibration.cache";
4443

45-
auto calibrator = trtorch::ptq::make_int8_calibrator(std::move(calibration_dataloader), calibration_cache_file, false);
44+
auto calibrator = trtorch::ptq::make_int8_calibrator(std::move(calibration_dataloader), calibration_cache_file, true);
4645
//auto calibrator = trtorch::ptq::make_int8_cache_calibrator(calibration_cache_file);
4746

4847

0 commit comments

Comments
 (0)