Open Pyf1234 opened 1 year ago
Hello, I found that the tensor in manolayer is not moved to "cuda". I guess it couldn't run on GPU. Is there any implementation which could run on GPU ? Thanks a lot!
Hello, I found that the tensor in manolayer is not moved to "cuda". I guess it couldn't run on GPU. Is there any implementation which could run on GPU ? Thanks a lot!