mirror of
https://github.com/graphdeco-inria/gaussian-splatting
synced 2025-04-08 06:54:06 +00:00
Merge pull request #30 from DerThomy/main
Empty torch cache after optimizer tensor replacement
This commit is contained in:
commit
7b4fc9f76a
@ -442,6 +442,8 @@ class GaussianModel:
|
||||
self._scaling = optimizable_tensors["scaling"]
|
||||
self._rotation = optimizable_tensors["rotation"]
|
||||
|
||||
torch.cuda.empty_cache()
|
||||
|
||||
return optimizable_tensors
|
||||
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user