Training device?

[mariolew]

Hi, I'd like to re-train this model on my own data, however, out of memory error occurs even samples_per_gpu is set to 1. I'm using gtx 2080ti.

[Annbless]

Hi,

ViTPose has a smaller training memory than HRNet-w48 when inputting 256x192 images. Perhaps you could try resizing the input images for your own data to alleviate the training memory problem.

[mariolew]

I already use 256x192

[KeyKy]

I also get out of memory on V100 with batchsize64;