While trying to export a SSD model to INT8, it is required to generate the cal.tensorfile
so I followed the docs and sample notebooks and used below command to generate:
!tlt-int8-tensorfile ssd \
--cal_image_dir /dataset/images \
--cal_data_file exported_model/cal.tensorfile
Error:
Using TensorFlow backend.
2021-03-17 07:21:00.029695: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
WARNING:tensorflow:Deprecation warnings have been disabled. Set TF_ENABLE_DEPRECATION_WARNINGS=1 to re-enable them.
Traceback (most recent call last):
File "/usr/local/bin/tlt-int8-tensorfile", line 8, in <module>
sys.exit(main())
File "/home/vpraveen/.cache/dazel/_dazel_vpraveen/216c8b41e526c3295d3b802489ac2034/execroot/ai_infra/bazel-out/k8-fastbuild/bin/magnet/packages/iva/build_wheel.runfiles/ai_infra/iva/common/magnet_tensorfile.py", line 41, in main
ImportError: cannot import name 'calibration_tensorfile'
Any alternate method to export SSD model in INT8 will be helpful.
Thanks