58 lines
3.3 KiB
Plaintext
58 lines
3.3 KiB
Plaintext
----------------------------------------------------------------
|
|
Layer (type) Output Shape Param #
|
|
================================================================
|
|
Conv2d-1 [-1, 32, 224, 224] 10,400
|
|
BatchNorm2d-2 [-1, 32, 224, 224] 64
|
|
Conv2d-3 [-1, 64, 112, 112] 32,832
|
|
BatchNorm2d-4 [-1, 64, 112, 112] 128
|
|
Conv2d-5 [-1, 128, 56, 56] 131,200
|
|
BatchNorm2d-6 [-1, 128, 56, 56] 256
|
|
Conv2d-7 [-1, 128, 56, 56] 147,584
|
|
BatchNorm2d-8 [-1, 128, 56, 56] 256
|
|
Conv2d-9 [-1, 128, 56, 56] 147,584
|
|
BatchNorm2d-10 [-1, 128, 56, 56] 256
|
|
ResidualBlock-11 [-1, 128, 56, 56] 0
|
|
Conv2d-12 [-1, 128, 56, 56] 147,584
|
|
BatchNorm2d-13 [-1, 128, 56, 56] 256
|
|
Conv2d-14 [-1, 128, 56, 56] 147,584
|
|
BatchNorm2d-15 [-1, 128, 56, 56] 256
|
|
ResidualBlock-16 [-1, 128, 56, 56] 0
|
|
Conv2d-17 [-1, 128, 56, 56] 147,584
|
|
BatchNorm2d-18 [-1, 128, 56, 56] 256
|
|
Conv2d-19 [-1, 128, 56, 56] 147,584
|
|
BatchNorm2d-20 [-1, 128, 56, 56] 256
|
|
ResidualBlock-21 [-1, 128, 56, 56] 0
|
|
Conv2d-22 [-1, 128, 56, 56] 147,584
|
|
BatchNorm2d-23 [-1, 128, 56, 56] 256
|
|
Conv2d-24 [-1, 128, 56, 56] 147,584
|
|
BatchNorm2d-25 [-1, 128, 56, 56] 256
|
|
ResidualBlock-26 [-1, 128, 56, 56] 0
|
|
Conv2d-27 [-1, 128, 56, 56] 147,584
|
|
BatchNorm2d-28 [-1, 128, 56, 56] 256
|
|
Conv2d-29 [-1, 128, 56, 56] 147,584
|
|
BatchNorm2d-30 [-1, 128, 56, 56] 256
|
|
ResidualBlock-31 [-1, 128, 56, 56] 0
|
|
ConvTranspose2d-32 [-1, 64, 113, 113] 131,136
|
|
BatchNorm2d-33 [-1, 64, 113, 113] 128
|
|
ConvTranspose2d-34 [-1, 32, 225, 225] 32,800
|
|
BatchNorm2d-35 [-1, 32, 225, 225] 64
|
|
ConvTranspose2d-36 [-1, 32, 225, 225] 82,976
|
|
Dropout-37 [-1, 32, 225, 225] 0
|
|
Conv2d-38 [-1, 1, 224, 224] 129
|
|
Dropout-39 [-1, 32, 225, 225] 0
|
|
Conv2d-40 [-1, 1, 224, 224] 129
|
|
Dropout-41 [-1, 32, 225, 225] 0
|
|
Conv2d-42 [-1, 1, 224, 224] 129
|
|
Dropout-43 [-1, 32, 225, 225] 0
|
|
Conv2d-44 [-1, 1, 224, 224] 129
|
|
================================================================
|
|
Total params: 1,900,900
|
|
Trainable params: 1,900,900
|
|
Non-trainable params: 0
|
|
----------------------------------------------------------------
|
|
Input size (MB): 0.77
|
|
Forward/backward pass size (MB): 219.96
|
|
Params size (MB): 7.25
|
|
Estimated Total Size (MB): 227.97
|
|
----------------------------------------------------------------
|