Structure | Parameter |
Conv_1 (Leaky_relu) | in_channels = 1, out_channels = 4, kernel = (3*3), stride = (1, 1), padding = 1 |
Maxpool_1 | kernel = (2*2), stride = (2, 2), padding = 0 |
Conv_2 (Leaky_relu) | in_channels = 4, out_channels = 8, kernel = (3*3), stride = (1, 1), padding = 1 |
Maxpool_2 | kernel = (2*2), stride = (2, 2), padding = 0 |
F. interpolate | scale_factor = 2, mode = 'nearest' |
Conv_3 (Leaky_relu) | in_channels = 8, out_channels = 4, kernel = (3*3), stride = (1, 1), padding = 1 |
F.interpolate | scale_factor = 2, mode = 'nearest' |
Conv_4 (Leaky_relu) | in_channels = 4, out_channels = 1, kernel = (3*3), stride = (1, 1), padding = 1 |