Structure | Parameter |
Conv_5 | in_channels = 16, out_channels = 32, kernel = (3*3), stride = (1, 1), padding = 1 |
Conv_6 (Leaky_relu) | in_channels = 32, out_channels = 64, kernel = (3*3), stride = (2, 2), padding = 1 |
Linear_1 (Leaky_relu) | input = 576, output = 128 |
Linear_2 | input = 128, output = 3 |