Add cache cleaning

demo/gs_train.py

@@ -290,5 +290,5 @@ def train(
         return renders_path
     
     renders_path = render_path(dataset, opt.iterations, pipe, render_resize_method='crop')
-
+    torch.cuda.empty_cache()
     return renders_path, point_cloud_path

demo/mast3r_demo.py

@@ -239,6 +239,7 @@ def get_reconstructed_scene(image_size, current_scene_state,
     print(f"colmap_data_dir: {colmap_data_dir}")
     print(f"outfile_name: {outfile_name}")
     print(f"cache_dir: {cache_dir}")
+    torch.cuda.empty_cache()
     return scene_state, outfile