Method train_mode

text2motion/trainers/ddpm_trainer.py:150–151

(self)

        self.encoder = self.encoder.to(device)

    def train_mode(self):
        self.encoder.train()

    def eval_mode(self):
        self.encoder.eval()