cache

visualcloze.py

@@ -430,6 +430,8 @@ class VisualClozeModel:
                 output_image = to_pil_image(row_sample.float())
                 output_images.append(output_image)
             
+            torch.cuda.empty_cache()
+            
             ret = []
             ret_w = output_images[-1].width
             ret_h = output_images[-1].height