...
Note: If the user wishes to use the Gluon vision models, it's necessary to install the gluoncv
pip package:
Code Block |
---|
pip install gluoncv |
...
The above package is based on a separate repository.
For Gluon models specifically, we need to add a data symbol to the model to load the data, as well as apply the softmax layer, because the Gluon models only present the logits that are to be presented for softmax. This is shown in python ${MXNET_HOME}/tests/python/tensorrt/test_tensorrt_resnet_resnext.py
. Here's the relevant code:
...