Class TinyModel

tests/callbacks_test.py:27–32

class TinyModel(torch.nn.Module):
    def __init__(self):
        """Initialize a tiny torch module for callback tests."""
        super(TinyModel, self).__init__()
        self.linear = torch.nn.Linear(1, 1)
        self.stop_training = False


def test_callback_and_callback_list_flow():