running on GPU #29
Description
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!
{{ message }}
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!