tensorrt-int8.py 170 Bytes
Newer Older
limm's avatar
limm committed
1
2
3
4
5
_base_ = ['./tensorrt.py']

backend_config = dict(common_config=dict(fp16_mode=True, int8_mode=True))

calib_config = dict(create_calib=True, calib_file='calib_data.h5')