(self, opt: Options, **model_kwargs)
| 12 | |
| 13 | class SplatModel(StaticEncoder): |
| 14 | def __init__(self, opt: Options, **model_kwargs): |
| 15 | super().__init__(opt) |
| 16 | self.opt = opt |
| 17 | self.model = SplatPredictor(opt, **model_kwargs) |
| 18 | if hasattr(opt, 'compile') and opt.compile: |
| 19 | self.model = torch.compile(self.model) |
| 20 | self.gaussian_renderer = gaussian_renderer_dynamic.render |
| 21 | self.background = torch.tensor(opt.background_color, dtype=torch.float32, device="cuda") |
| 22 | self.lpips_loss = None |
| 23 | |
| 24 | def load_state_dict(self, state_dict, strict=True): |
| 25 | # if opt.use_dino, remove missing keys related to condition_encoder |
nothing calls this directly
no test coverage detected