Open umgpy opened 3 years ago
The main part of the algorithm is implemented with CUDA. Only cuda_op_kernel.cu.cc, which is for connecting the cuda part and the python part, is using tensorflow. Pytorch can read the .so file but not I haven't tried whether pytorch has corresponding functions to replace the tf functions in cuda_op_kernel.cu.cc.
Hi I was looking for a PyTorch implementation of the project. I'm willing to translate the code myself if there is no implementation. Just wanted to know how can I go about it. Cheers