Thank you for your work. I am very interested in your work and tried to run it on my GPU.
I have tried to modify your code for multi-gpu training. However, I got :
RuntimeError: The size of tensor a (8) must match the size of tensor b (16) at non-singleton dimension 0
But I've used the same way to enable multi-gpu training in my other projects and they all worked well. It seems there may be some problems with your code. Could you kindly provide a version that can be run on multiple gpus? Thank you so much.
Thank you for your work. I am very interested in your work and tried to run it on my GPU. I have tried to modify your code for multi-gpu training. However, I got :
RuntimeError: The size of tensor a (8) must match the size of tensor b (16) at non-singleton dimension 0But I've used the same way to enable multi-gpu training in my other projects and they all worked well. It seems there may be some problems with your code. Could you kindly provide a version that can be run on multiple gpus? Thank you so much.