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