pytorch 训练过程内存泄露/显存泄露debug记录:dataloader和dataset导致的泄露
背景微调mask-rcnn代码,用的是torchvision.models.detection.maskrcnn_resnet50_fpn代码,根据该代码的注释,输入应该是:images,targets=None(List[Tensor],Optional[List[Dict[str,Tensor]]])->Tuple[Dict[str,Tensor],List[Dict[str,Tensor]]