def METHOD_NAME(self):
    for _batch_idx, (inputs, targets) in enumerate(self.trainloader):
        inputs, targets = inputs.to(self.device), targets.to(self.device)
        self.optimizer.zero_grad()
        outputs = self.model(inputs)
        loss = self.criterion(outputs, targets)
        loss.backward()
        self.optimizer.step()