Skip to content

Commit

Permalink
Fix #22, TRT8 AttributeError: 'tensorrt.tensorrt.Builder' object has …
Browse files Browse the repository at this point in the history
…no attribute 'int8_calibrator' (#23)

* Fix #22, AttributeError: 'tensorrt.tensorrt.Builder' object has no attribute 'int8_calibrator' in TRT8

* undo
  • Loading branch information
v-qjqs authored Aug 29, 2021
1 parent 8cf19f2 commit 6659898
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion torch2trt_dynamic/torch2trt_dynamic.py
Original file line number Diff line number Diff line change
Expand Up @@ -584,7 +584,7 @@ def torch2trt_dynamic(module,
config.set_calibration_profile(profile)
if version.parse(trt.__version__) < version.parse('8'):
builder.int8_mode = int8_mode
builder.int8_calibrator = config.int8_calibrator
builder.int8_calibrator = config.int8_calibrator

engine = builder.build_engine(network, config)

Expand Down

0 comments on commit 6659898

Please sign in to comment.