I use the command below.
tlt-export detectnet_v2 -m /workspace/tlt-experiments/detectnet_v2/weights/resnet_18retrain.tlt -o /workspace/tlt-experiments/detectnet_v2/weights/resnet18_int8.etlt -k NHRvZzAwbHFncTk0MXJ0YmwwbXB1bGxhbnU6MjYzNzc2MDctYzQ5MC00NjkxLThkODAtODM0NDc3ZTRhNTNh --cal_data_file /workspace/tlt-experiments/detectnet_v2/weights/calibration.tensor --data_type int8 --batches 8 --max_batch_size 8 --cal_cache_file /workspace/tlt-experiments/detectnet_v2/weights/calibration_cache.bin --engine_file /workspace/tlt-experiments/detectnet_v2/weights/resnet_18trt.engine
Tensorfile conversion command is
tlt-int8-tensorfile detectnet_v2 -e /workspace/tlt-experiments/detectnet_v2/detectnet_v2_prune_resnet18_kitti.txt -o /workspace/tlt-experiments/detectnet_v2/weights/calibration.tensor -m 8
tensorfile was converted successfully.
The error is at tlt-export command.
Using TensorFlow backend. NOTE: UFF has been tested with TensorFlow 1.14.0. WARNING: The version of TensorFlow installed on this system is not guaranteed to work with UFF. DEBUG [/usr/lib/python2.7/dist-packages/uff/converters/tensorflow/converter.py:96] Marking ['output_cov/Sigmoid', 'output_bbox/BiasAdd'] as outputs [TensorRT] INFO: Detected 1 inputs and 2 output network tensors. [TensorRT] WARNING: Current optimization profile is: 0. Please ensure there are no enqueued operations pending in this context prior to switching profiles [TensorRT] INFO: Starting Calibration with batch size 16. DEPRECATED: This variant of get_batch is deprecated. Please use the single argument variant described in the documentation instead. Traceback (most recent call last): File "/usr/local/bin/tlt-export", line 8, in <module> sys.exit(main()) File "./common/export/app.py", line 234, in main File "./common/export/base_exporter.py", line 411, in export File "./modulus/export/_tensorrt.py", line 515, in __init__ File "./modulus/export/_tensorrt.py", line 414, in __init__ File "./modulus/export/_tensorrt.py", line 104, in get_batch ValueError: Data file batch size (4) < request batch size (16)
My another query is why tensorflow version is not 14.1 in docker. My docker is tlt-2.