diff --git a/scene/gaussian_model.py b/scene/gaussian_model.py
index f611338..e9fd1ab 100644
--- a/scene/gaussian_model.py
+++ b/scene/gaussian_model.py
@@ -442,6 +442,8 @@ class GaussianModel:
         self._scaling = optimizable_tensors["scaling"]
         self._rotation = optimizable_tensors["rotation"] 
 
+        torch.cuda.empty_cache()
+        
         return optimizable_tensors