modules.conv.encnet*¶
Function ยท nn.Module
net = encnet*(
order=2, kernel_size=3, in_planes=1, out_length=2
)
Instant presents of mdnc.module.conv.EncoderNet*d
.
Arguments¶
Requries
Argument | Type | Description |
---|---|---|
order | int | The order of the convolutional layers, could be 1 , 2 , or 3 . |
kernel_size | int or(int,) | The kernel size of each convolutional layer. |
in_planes | int | The channel number of the input data. |
out_length | int | The length of the output vector, if not set, the output would not be flattened. |
APIs¶
API | Net depth | Channel | Layer configs | Source |
---|---|---|---|---|
encnet12 | 5 | 64 | [2, 2, 2, 2, 2] | |
encnet15 | 5 | 64 | [2, 2, 3, 3, 3] | |
encnet17 | 5 | 64 | [3, 3, 3, 3, 3] | |
encnet22 | 5 | 64 | [4, 4, 4, 4, 4] |
Last update: March 14, 2021