cleanup

nerfstudio-project · Oct 29, 2024 · 834e4e8 · 834e4e8
1 parent c8fd74d
commit 834e4e8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/examples/blur_kernel.py → examples/blur_opt.py b/examples/blur_kernel.py → examples/blur_opt.py
diff --git a/examples/simple_trainer.py b/examples/simple_trainer.py
@@ -28,14 +28,14 @@
 from fused_ssim import fused_ssim
 from torchmetrics.image.lpip import LearnedPerceptualImagePatchSimilarity
 from typing_extensions import Literal, assert_never
+from blur_opt import BlurOptModule
 from utils import AppearanceOptModule, CameraOptModule, knn, rgb_to_sh, set_random_seed
 from lib_bilagrid import (
     BilateralGrid,
     slice,
     color_correct,
     total_variation_loss,
 )
-from blur_kernel import BlurOptModule
 
 from gsplat.compression import PngCompression
 from gsplat.distributed import cli
@@ -412,7 +412,7 @@ def __init__(
 
         self.blur_optimizers = []
         if cfg.blur_opt:
-            self.blur_module = BlurOptModule(cfg, len(self.trainset)).to(self.device)
+            self.blur_module = BlurOptModule(len(self.trainset)).to(self.device)
             self.blur_module.zero_init()
             self.blur_optimizers = [
                 torch.optim.Adam(