load NGC models in native tensorflow

[hchintada]

Hi,

I need to load the NGC DenseNet121 pre-trained model into native TF.
link to the model: https://ngc.nvidia.com/catalog/models/nvidia:med:clara_xray_classification_chest_amp

Loading the checkpoints in the 'models' directory into densenet121 architecture with model.load_weights('path') is throwing some objects had attributes which were not restored error

TF version 2.1.
Please suggest the correct approach.