def __init__()

in ViT4MNIST/vit_pytorch.py [0:0]

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

    def __init__(self, dim, hidden_dim):
        super().__init__()
        self.net = nn.Sequential(
            nn.Linear(dim, hidden_dim),
            nn.GELU(),
            nn.Linear(hidden_dim, dim)
        )