A Error in training about spdcn-cac

[Elin24]

This happened because the point locations may be out of range. I add the clip() function to avoid it, as shown in line-56:

points[:, 1] = torch.clip(points[:, 1], min=0, max=h-1)
points[:, 0] = torch.clip(points[:, 0], min=0, max=w-1)

[ytddd75]

Thanks for your thoughtful reply, the problem has been solved.
By the way, dose the code you provide allow for multi-GPU training?

[Elin24]

I'm afraid the answer is no.

However, the batch size is 16, and the memory requirement is about 21G, so one 3090-GPU is enough to train this model. You can reduce the batch size and adjust the learning rate accordingly for training with limited resources.

Another way to reduce computation costs is using FP16, which may decrease the accuracy a little but save lots of memory.

[ytddd75]

Thanks for your thoughtful reply.