(self)
| 21 | self._freeze_detector() |
| 22 | |
| 23 | def _freeze_detector(self): |
| 24 | |
| 25 | self.detector = [ |
| 26 | self.backbone, self.neck, self.rpn_head, self.roi_head.bbox_head |
| 27 | ] |
| 28 | for model in self.detector: |
| 29 | model.eval() |
| 30 | for param in model.parameters(): |
| 31 | param.requires_grad = False |
| 32 | |
| 33 | def prepare_cfg(self, kwargs): |
| 34 | if kwargs.get('train_cfg', False): |