(self)
| 482 | raise NotImplementedError("Currently only support probabilistic predictor") |
| 483 | |
| 484 | def _freeze(self): |
| 485 | for name, param in self.named_parameters(): |
| 486 | if "xyz_static" in name: |
| 487 | param.requires_grad = True |
| 488 | else: |
| 489 | param.requires_grad = False |
| 490 | |
| 491 | def forward_encoder(self, frames, depths, cond_times=None): |
| 492 | mask = torch.ones_like(depths) # dummy mask |
nothing calls this directly
no outgoing calls
no test coverage detected