Convert the model into training mode while keep layers freezed.
(self, mode=True)
| 590 | return outs |
| 591 | |
| 592 | def train(self, mode=True): |
| 593 | """Convert the model into training mode while keep layers freezed.""" |
| 594 | super(FocalNet, self).train(mode) |
| 595 | self._freeze_stages() |
| 596 | |
| 597 | |
| 598 | class D2FocalNet(FocalNet, Backbone): |
no test coverage detected