vision/amazon-sagemaker-pytorch-detectron2/container_training/sku-110k/evaluation/factory.py (1 line): - line 13: """TODO add doc""" sagemaker/distributed-training/train_pytorch_single_maskrcnn.py (1 line): - line 175: torch.cuda.empty_cache() # TODO check if it helps sagemaker/distributed-training/train_pytorch_smdataparallel_maskrcnn.py (1 line): - line 192: torch.cuda.empty_cache() # TODO check if it helps