Convert the model into training mode while keep layers freezed.
(self, mode=True)
| 489 | return tuple(outs) |
| 490 | |
| 491 | def train(self, mode=True): |
| 492 | """Convert the model into training mode while keep layers freezed.""" |
| 493 | super(MultiModalSwinTransformer, self).train(mode) |
| 494 | self._freeze_stages() |
| 495 | |
| 496 | |
| 497 | class MMBasicLayer(nn.Module): |