search

TencentYoutuResearch / CrowdCounting-P2PNet

The official codes for the ICCV2021 Oral presentation "Rethinking Counting and Localization in Crowds: A Purely Point-Based Framework"
Other
419 stars 113 forks source link

How to resolve "out of memory" during evaluate on high resolution single image? #17

Closed Tracyummy closed 2 years ago

Tracyummy commented 2 years ago

https://github.com/TencentYoutuResearch/CrowdCounting-P2PNet/blob/5c91a81ca062b1c7fd3db3ad1c55b1c21f0a7455/train.py#L187 As I run code above, an error "out of memory" pops while I run the script on single gpu RTX6000 with 24G memory.

jhw55 commented 2 years ago

Are you solved the problem?I had meet it,too.