Method load_state_dict

model/splat_model.py:57–61

Load state dict, filtering out non-trainable modules if present.

(self, state_dict, strict=True)

        return state_dict
    
    def load_state_dict(self, state_dict, strict=True):
        """Load state dict, filtering out non-trainable modules if present."""
        filtered_state_dict = {k: v for k, v in state_dict.items() 
                             if not k.startswith('lpips_loss') and not k.startswith('tracker_prior')}
        return super().load_state_dict(filtered_state_dict, strict=strict)
        
    def forward_gaussians(self, frames, depths, cond_times=None):
        # frames: [B, V, C, H, W]