diff --git a/examples/blur_kernel.py b/examples/blur_opt.py similarity index 100% rename from examples/blur_kernel.py rename to examples/blur_opt.py diff --git a/examples/simple_trainer.py b/examples/simple_trainer.py index 8e369679..2e1eaa77 100644 --- a/examples/simple_trainer.py +++ b/examples/simple_trainer.py @@ -28,6 +28,7 @@ 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, @@ -35,7 +36,6 @@ 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(