Retail Object Detection with EfficientDet - encryption key error

• 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

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.