    def set_train_mode(self, model: nn.Module):
        """
        Convert the model to training mode.

        Args:
            model (`nn.Module`):
        """
        model.train()
        if self.configs.freeze_bn:
            model.apply(freeze_bn)
Convert the model to training mode.

Args:
model (`nn.Module`):
towhee/trainer/trainer.py