def METHOD_NAME(self):
    model = self.model
    (images, ) = self.example_inputs
    model.METHOD_NAME()
    targets = self.example_target
    output = model(images)
    loss = self.criterion(output, targets)
    loss.backward()
    self.optimizer.step()
    self.optimizer.zero_grad()