ubc-vision / COTR

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

Memory footprints #41

Open DK-HU opened 2 years ago

DK-HU commented 2 years ago

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

jiangwei221 commented 2 years ago

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