Wuwenll

It seems that there are only pre-trained models in “download_pretrained_models.sh”. Can you share your trained models?

how to use gpu when testing？

2

According to the content of the readme, I successfully run your method, but the default is to use CPU. How can use a GPU when testing？