From 289d3435d5d46e4622605100f61487dcd62a76f0 Mon Sep 17 00:00:00 2001 From: abh3067 <106952493+abh3067@users.noreply.github.com> Date: Wed, 5 Mar 2025 00:49:36 +0530 Subject: [PATCH] Update main.py --- Segmentation/main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Segmentation/main.py b/Segmentation/main.py index 7603fea..8a12a2d 100644 --- a/Segmentation/main.py +++ b/Segmentation/main.py @@ -409,7 +409,7 @@ def download_checkpoints(model_name): # target_height=args.crop_height, target_width=args.crop_width).eval(session=sess) # ***** THIS CAUSES A MEMORY LEAK AS NEW TENSORS KEEP GETTING CREATED ***** - # memory() + # memory()ffff if args.batch_size == 1: input_image_batch = input_image_batch[0]