Method train

model/encoder_model.py:56–60

(self, mode=True)

        return super().load_state_dict(filtered_state_dict, strict=False)
    
    def train(self, mode=True):
        super().train(mode)
        if 'lpips_loss' in self.__dict__:
            self.lpips_loss.eval()
        return self
    
    def forward_gaussians(self, frames, depths, cond_times=None):
        # frames: [B, V, C, H, W]