uber-research / UPSNet

UPSNet: A Unified Panoptic Segmentation Network
Other
649 stars 120 forks source link

OSError: [Errno 12] Cannot allocate memory #93

Closed renxuezi closed 5 years ago

renxuezi commented 5 years ago

How to remove all multi-thread code and use a single core in base_dataset.py? I have changed "annotations.extend(p[0]);pan_all.extend(p[1])" into "pan_all.extend(p)", now it got stuck when the results were output,how should I modify code?

YuwenXiong commented 5 years ago

You can set cpu_num variable in base_dataset.py to 1 to avoid memory issue.

renxuezi commented 5 years ago

Thank you for your reply,my problem solved after I've changed to a better GPU.