6 lines
171 B
Python
6 lines
171 B
Python
|
_base_ = ['./tensorrt.py']
|
||
|
|
|
||
|
|
create_calib = True
|
||
|
|
calib_params = dict(calib_file='calib_data.h5')
|
||
|
|
tensorrt_params = dict(shared_params=dict(fp16_mode=True, int8_mode=True))
|