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