-
Notifications
You must be signed in to change notification settings - Fork 109
Open
Description
I found out that memory leak may happen in box_list_iou (https://github.com/zhangxiaosong18/FreeAnchor/blob/master/maskrcnn_benchmark/modeling/rpn/free_anchor_loss.py#L108)
when it processes large number of ground truth, of course the gpu memory grows up. but it never frees the memory it holds.
turning to cpu mode is a workaround but very slow.
Metadata
Metadata
Assignees
Labels
No labels