Description
I am trying to train a ReIdentificationNet Transformer model, based on the Swin transformer architecture. However, I can’t find a suitable pretrained model in the model repository.
I am using the following base image: nvcr.io/nvidia/tao/tao-toolkit:5.2.0-pyt2.1.0
Then running the following command for getting the list of models:
ngc registry model list nvidia/tao/* --format_type=csv > model_zoo.txt
This is the output: model_zoo.txt (20.8 KB)
Where should I search for the model weights?
2 posts - 2 participants