def METHOD_NAME(self):
    """Setup the model and optimizer which are used in every test method.
    TestCase calls functions in this order: setUp() -> testMethod() ->
    tearDown() -> cleanUp()
    """
    self.model = ToyModel()
    self.optimizer = optim.SGD(
        self.model.parameters(), lr=0.05, momentum=0.01, weight_decay=5e-4)