Layer | Output channels | Output dimension | Kernel, stride, padding | Parameters | Activation |
---|---|---|---|---|---|
Input | 1 | [64,64] | – | – | – |
Convolution-2D | 64 | [32,32] | [4,4], 2, 1 | 1024 | LeakyReLU |
Convolution-2D | 128 | [16,16] | [4,4], 2, 1 | 131,072 | – |
BatchNorm-2D | 128 | [16,16] | – | 256 | LeakyReLU |
Convolution-2D | 256 | [8,8] | [4,4], 2, 1 | 524,288 | – |
BatchNorm-2D | 256 | [8,8] | – | 512 | LeakyReLU |
Convolution-2D | 1 | [5,5] | [4,4], 1, 0 | 4,096 | Sigmoid |
Output | 1 | 1 | – | – | – |