Update apps/infer.py

apps/infer.py

@@ -657,13 +657,13 @@ def generate_model(in_path, fitting_step=50):
     overlap_path = img_overlap_path
     vis_tensor_path = osp.join(out_dir, cfg.name, f"vid/{data['name']}_in_tensor.pt")
 
-    # # clean all the variables
-    # for element in dir():
-    #     if 'path' not in element:
-    #         del locals()[element]
-
-    # import gc
-    # gc.collect()
-    # torch.cuda.empty_cache()
+    # clean all the variables
+    for element in dir():
+        if 'path' not in element:
+            del locals()[element]
+
+    import gc
+    gc.collect()
+    torch.cuda.empty_cache()
 
     return [smpl_glb_path, refine_glb_path, overlap_path, vis_tensor_path]