search

SysCV / transfiner

Mask Transfiner for High-Quality Instance Segmentation, CVPR 2022
https://www.vis.xyz/pub/transfiner
Apache License 2.0
528 stars 60 forks source link

Excessive memory usage during inference #35

Closed liclshixiaokeaiya closed 2 years ago

liclshixiaokeaiya commented 2 years ago

When I use the trained model for inference, there will be a problem that a certain image occupies too much GPU memory. The peak may reach more than 10G.

lkeab commented 2 years ago

You can slightly reduce the number of limit here if you have a tight memory.

liclshixiaokeaiya commented 2 years ago

Do I need to retrain after reduce limit ?

lkeab commented 2 years ago

No.