Method optimizer_fn

gsplat/strategy/STG_Strategy.py:353–354

(key: str, v: Tensor)

            else:
                raise ValueError(f"Unexpected parameter name: {name}")
        def optimizer_fn(key: str, v: Tensor) -> Tensor:
            return v

        # update the parameters and the state in the optimizers
        _update_param_with_optimizer(param_fn, optimizer_fn, params, optimizers, names=["omega"])