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]