def _run_batch()

in pai-python-sdk/training/pytorch_ddp/train_src/train_multinode.py [0:0]

• 6 lines of code
• 1 McCabe index (conditional complexity)

    def _run_batch(self, source, targets):
        self.optimizer.zero_grad()
        output = self.model(source)
        loss = F.cross_entropy(output, targets)
        loss.backward()
        self.optimizer.step()