Skip to content

Instantly share code, notes, and snippets.

@rmccorm4

rmccorm4/builder_config_migration.py

Created May 10, 2020 23:12
Show Gist options
  • Save rmccorm4/a4f7275ff43f411fae6ef7e03f06050e to your computer and use it in GitHub Desktop.
Save rmccorm4/a4f7275ff43f411fae6ef7e03f06050e to your computer and use it in GitHub Desktop.
Download ZIP
Raw
builder_config_migration.py
# Previously: builder.fp16 = True
config.set_flag(trt.BuilderFlag.FP16)
# Previously: builder.int8 = True
config.set_flag(trt.BuilderFlag.INT8)
# Previously: builder.int8_calibrator = MyCustomCalibrator()
config.int8_calibrator = MyCustomCalibrator()
# ...
# Previously: builder.build_cuda_engine(network)
builder.build_engine(network, config)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment