Can this project train with GTX3090 (24G)? I met out of memory

xingyizhou / GTR

Global Tracking Transformers, CVPR 2022

375 stars 57 forks source link

Can this project train with GTX3090 (24G)? I met out of memory #40

Open dongfengxijian opened 2 years ago

dongfengxijian commented 2 years ago

Hello, Thank you for sharing the excellent project!

My lab don't have RTX 6000GPUs (24G, referenced in paper ) but some GTX 3090 GPUs (24G). So I want to train GTR with GTX3090 on the MOT dataset. Howevr, I met out of memory error even with batchsize = 1. Could you give me any idea?

Project627 commented 1 year ago

i met the same problem with the same device..

Project627 commented 1 year ago

make the train size smaller will be ok