Merge pull request #30 from DerThomy/main

Empty torch cache after optimizer tensor replacement
ubc-vision · Jan 2, 2025 · 7b4fc9f · 7b4fc9f
2 parents 491e17a + 389bbe4
commit 7b4fc9f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/scene/gaussian_model.py b/scene/gaussian_model.py
@@ -442,6 +442,8 @@ def replace_tensors_to_optimizer(self, inds=None):
         self._scaling = optimizable_tensors["scaling"]
         self._rotation = optimizable_tensors["rotation"] 
 
+        torch.cuda.empty_cache()
+
         return optimizable_tensors