Riva ner model

AI & Data Science Deep Learning (Training & Inference) Riva
1

Hardware - Orin
Operating System - JP 5.0/Ubuntu 20.04
Riva Version - 2.2.1

I am trying to use the available Riva_ner model for token classification. I downloaded the model from NGC. I started the RIVA docker container. When I tried to use the model, I get an error that says it is not a RIVA_API model. Do I need to change any config parameter so that the model gets loaded in the Docker container?

Thank you.

2

Hi @raghavendra.ramya

Thanks for your interest in Riva,

Can we try or are we trying instructions in the below link
https://docs.nvidia.com/deeplearning/riva/user-guide/docs/nlp/nlp-customizing.html

Thanks

3

Thank you for your response. Yes, I am following the instructions in the link. I have not been able to load the model for ner on the Riva docker.