FOD icon indicating copy to clipboard operation
FOD copied to clipboard

A bug in the project.

Open Delta0406 opened this issue 1 year ago • 1 comments

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.

Delta0406 avatar Aug 14 '23 02:08 Delta0406