Skip to content

Commit

Permalink
fix(//cpp/ptq): remove some logging from ptq app
Browse files Browse the repository at this point in the history
Signed-off-by: Naren Dasan <naren@narendasan.com>
Signed-off-by: Naren Dasan <narens@nvidia.com>
  • Loading branch information
narendasan committed Apr 24, 2020
1 parent 825be69 commit b989c7f
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions cpp/ptq/main.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,6 @@
#include <sys/stat.h>

int main(int argc, const char* argv[]) {
trtorch::logging::set_reportable_log_level(trtorch::logging::Level::kERROR);
if (argc < 3) {
std::cerr << "usage: ptq <path-to-module> <path-to-cifar10>\n";
return -1;
Expand Down Expand Up @@ -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);


Expand Down

0 comments on commit b989c7f

Please sign in to comment.