ubc-vision / COTR

Code release for "COTR: Correspondence Transformer for Matching Across Images"(ICCV 2021)
Apache License 2.0
460 stars 58 forks source link

Memory footprints #41

Open DK-HU opened 1 year ago

DK-HU commented 1 year ago

Could you please tell the memory footprints of COTR during traing process? Thanks

jiangwei221 commented 1 year ago

For first two stages, we load images into main memory, that’s about 64GB RAM. As for VRAM, that’s around 20GB.