Tlt v3.0 ssd export ImportError: cannot import name 'main'

Accelerated Computing Intelligent Video Analytics TAO Toolkit
1

I run tlt-export in tlt v2.0, it work,
but I pull new tlt-streamanalytics:v3.0-dp-py3 docker, run tlt-export ,throw error:
Traceback (most recent call last):
File “/usr/local/bin/tlt-export”, line 5, in
from iva.common.export.app import main
ImportError: cannot import name ‘main’

2

Please share the full command and log when you run tlt-export.

3

tlt-export ssd
-m /workspace/tlt-experiments/ssd/pruned_resnet18_1_7_1_retrain/weights/ssd_resnet18_epoch_080.tlt
-k XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
-o /workspace/tlt-experiments/ssd/pruned_resnet18_1_7_1_export/ssd_resnet18_epoch_80_pruned.etlt
-e /workspace/tlt-experiments/ssd/specs/ssd_retrain_resnet18_kitti_1_7_1.txt
–batch_size 16
–data_type fp32

Using TensorFlow backend.
2021-02-21 09:39:05.741869: 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-export”, line 5, in
from iva.common.export.app import main
ImportError: cannot import name ‘main’

4

Your command is not compatible with TLT 3.0. Please modify it. Please note that TLT 3.0 is different from TLT 2.0.
Refer to
https://docs.nvidia.com/metropolis/TLT/tlt-user-guide/text/migrating_v2.0_v3.0.html
https://docs.nvidia.com/metropolis/TLT/tlt-user-guide/text/object_detection/ssd.html#exporting-the-model
https://docs.nvidia.com/metropolis/TLT/tlt-user-guide/text/tlt_launcher.html