(self, mode=True)
| 560 | return state_dict |
| 561 | |
| 562 | def train(self, mode=True): |
| 563 | super().train(mode) |
| 564 | if self.opt.use_dino: |
| 565 | self.condition_encoder.eval() |
| 566 | self.gs_predictor.eval() |
| 567 | return self |
| 568 | |
| 569 | def _freeze_predictor(self): |
| 570 | self.gs_predictor.eval() |
nothing calls this directly
no outgoing calls
no test coverage detected