clara-train-examples icon indicating copy to clipboard operation
clara-train-examples copied to clipboard

load NGC models in native tensorflow

Open hchintada opened this issue 3 years ago • 0 comments

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.

hchintada avatar Mar 08 '21 08:03 hchintada