From f2e908f7d9d26388cf2c6e4f06628e778b7d2ab0 Mon Sep 17 00:00:00 2001 From: waifairer Date: Thu, 19 Oct 2023 20:29:02 -0600 Subject: [PATCH] WIP --- examples/mask_rcnn/train.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/examples/mask_rcnn/train.py b/examples/mask_rcnn/train.py index b87eac689d5c4..3dfe596af117b 100644 --- a/examples/mask_rcnn/train.py +++ b/examples/mask_rcnn/train.py @@ -163,6 +163,12 @@ def simple(): img_metadata = coco.loadImgs(random_img_id)[0] img_filename = os.path.join(BASEDIR, 'train2017', img_metadata['file_name']) print("training", img_filename) + print_gpu_memory("before_load_no_transforms") + t = Tensor(Image.open(img_filename).convert("RGB"), requires_grad=True).realize() + print_gpu_memory("after_load_no_transforms") + del t + gc.collect() + print_gpu_memory("after_del") print_gpu_memory("before_load") t = Tensor(build_transforms()(Image.open(img_filename).convert("RGB")).numpy(), requires_grad=True).realize() print_gpu_memory("after_load")