class Flatten(nn.Module):
    def forward(self, input):
        return input.view(input.size(0), -1)

class ActionNet(Module):

    def __init__(self, num_class=4):
        super(ActionNet, self).__init__()
        
        self.cnn_layer = Sequential(
            #conv1
            Conv2d(in_channels=1, out_channels=32, kernel_size=1),
            PReLU(num_parameters=32),
            MaxPool2d(kernel_size=3),
            #conv2
            Conv2d(in_channels=32, out_channels=64, kernel_size=1),
            PReLU(num_parameters=64),
            MaxPool2d(kernel_size=3),
            Flatten(),
            Linear(576, 128),
            ReLU(inplace=True),
            Dropout(0.5),
            Linear(128, num_class)
        )
    
    def forward(self, x):

        x = self.cnn_layer(x)
        return x
        


HI here is my network please correct me if i’m wrong @ibro45