| 567 | return self |
| 568 | |
| 569 | def _freeze_predictor(self): |
| 570 | self.gs_predictor.eval() |
| 571 | for name, param in self.gs_predictor.named_parameters(): |
| 572 | if "dynamic" in name: |
| 573 | param.requires_grad = True |
| 574 | elif "encoder" in name: |
| 575 | param.requires_grad = False |
| 576 | elif "upsampler" in name: |
| 577 | param.requires_grad = False |
| 578 | elif "gs_layer" in name: |
| 579 | param.requires_grad = False |
| 580 | else: |
| 581 | param.requires_grad = False |
| 582 | |
| 583 | def forward_encoder(self, frames, depths, cond_times=None): |
| 584 | frames = torch.cat([frames, depths], dim=2) |