FOD
FOD copied to clipboard
A bug in the project.
I found a bug in the project when I wanted to run the project in cuda:2. “RuntimeError :Expected all tensors to be on the same device, but found at least two devices, cuda:2 and cuda:0!” occurred. On closer inspection, I found that the problem is at pos_embed.cuda() of PositionEmbedding.