File tree Expand file tree Collapse file tree 1 file changed +1
-2
lines changed Expand file tree Collapse file tree 1 file changed +1
-2
lines changed Original file line number Diff line number Diff line change 13
13
#include < sys/stat.h>
14
14
15
15
int main (int argc, const char * argv[]) {
16
- trtorch::logging::set_reportable_log_level (trtorch::logging::Level::kERROR );
17
16
if (argc < 3 ) {
18
17
std::cerr << " usage: ptq <path-to-module> <path-to-cifar10>\n " ;
19
18
return -1 ;
@@ -42,7 +41,7 @@ int main(int argc, const char* argv[]) {
42
41
43
42
std::string calibration_cache_file = " /tmp/vgg16_TRT_ptq_calibration.cache" ;
44
43
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 );
46
45
// auto calibrator = trtorch::ptq::make_int8_cache_calibrator(calibration_cache_file);
47
46
48
47
You can’t perform that action at this time.
0 commit comments