clara-train-examples
clara-train-examples copied to clipboard
load NGC models in native tensorflow
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.