(self, state_dict, strict=True)
| 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 |
| 26 | missing_keys, unexpected_keys = super().load_state_dict(state_dict, strict=strict) |
| 27 | if self.opt.use_dino: |
| 28 | missing_keys = [k for k in missing_keys if "condition_encoder" not in k] |
| 29 | return missing_keys, unexpected_keys |
| 30 | |
| 31 | def forward_gaussians(self, frames, depths, cond_times=None): |
| 32 | # frames: [B, V, C, H, W] |
nothing calls this directly
no outgoing calls
no test coverage detected