from torch import nn


class MyAwesomeModel(nn.Module):
    def __init__(self):
        super().__init__()

        self.cnn = nn.Sequential(nn.Conv2d(in_channels=1, out_channels=5, kernel_size=3),
                                nn.ReLU(),
                                nn.Conv2d(in_channels=5, out_channels=3, kernel_size=3, stride=2)
                                )
        self.fc = nn.Sequential(nn.Linear(432, 100),
                                nn.ReLU(),
                                nn.Linear(100,10),
                                nn.LogSoftmax(dim=1)
                                )

    def forward(self, x):
        x = self.cnn(x).view(x.size(0), -1)
        return self.fc(x)


[((8, 33, 8, 88), 'torch.nn.Conv2d', 'nn.Conv2d', (), '', False, 'from torch import nn\n'), ((9, 32, 9, 41), 'torch.nn.ReLU', 'nn.ReLU', ({}, {}), '()', False, 'from torch import nn\n'), ((10, 32, 10, 97), 'torch.nn.Conv2d', 'nn.Conv2d', (), '', False, 'from torch import nn\n'), ((12, 32, 12, 51), 'torch.nn.Linear', 'nn.Linear', ({(12, 42, 12, 45): '432', (12, 47, 12, 50): '100'}, {}), '(432, 100)', False, 'from torch import nn\n'), ((13, 32, 13, 41), 'torch.nn.ReLU', 'nn.ReLU', ({}, {}), '()', False, 'from torch import nn\n'), ((14, 32, 14, 49), 'torch.nn.Linear', 'nn.Linear', ({(14, 42, 14, 45): '100', (14, 46, 14, 48): '10'}, {}), '(100, 10)', False, 'from torch import nn\n'), ((15, 32, 15, 52), 'torch.nn.LogSoftmax', 'nn.LogSoftmax', (), '', False, 'from torch import nn\n')]