Retail Object Detection with EfficientDet - encryption key error

Accelerated Computing Intelligent Video Analytics TAO Toolkit
,
1

• Network Type:efficientdet-d5
• TLT Version: 5.5

Hi there,

I am trying to use Efficientdet-d5 pretrained model from Retail Object Detection Retail Object Detection | NVIDIA NGC.
But there is an issue when trying to load the file. TAO is not recognising the encryption key ‘nvidia_tlt’
With the following training spec:

train:
    checkpoint: "/workspace/tao-experiments/models/trainable_retail_object_detection_v1.1/efficientdet-d5_090.tlt"
….
encryption_key: 'nvidia_tlt'

I am getting the following error

File "<frozen cv.efficientdet.model.efficientdet_module>", line 58, in __init__
 File "<frozen cv.efficientdet.model.efficientdet_module>", line 117, in _load_pretrained_weights
 File "<frozen cv.efficientdet.utils.helper>", line 74, in decode_eff
 File "<frozen eff.core.file>", line 268, in get_handle
 File "<frozen eff.core.file>", line 209, in decrypt
 File "<frozen eff.core.file>", line -1, in check_decryption
PermissionError: The provided passphrase is invalid

I have tried the trainable model from v1.0 and it can load it correctly with the same encryption key.

Is there anything wrong on how the weights were encrypted?

Thanks

2

Can you confirm that you are using TAO5.5 docker to run training?
If yes, the training spec file is not needed to contain “encryption_key”. Because since TAO 5.0 version, the encryption is deprecated.

Reference spec file is in tao_tutorials/notebooks/tao_launcher_starter_kit/retail_object_detection/specs/train.yaml at main · NVIDIA/tao_tutorials · GitHub.

Suggest to use newer version of pretrained model which has .pth as extension.