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?
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?