modules.resnet.encnet*¶

Function · nn.Module

net = encnet*(
    order=2, kernel_size=3, in_planes=1, out_length=2
)

Instant presents of mdnc.module.resnet.EncoderNet*d.

Arguments¶

Requries

Argument	Type	Description
`order`	`int`	The order of the residual blocks, could be `1`, `2`, or `3`.
`kernel_size`	`int` or `(int,)`	The kernel size of each residual block.
`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.

API	Net depth	Channel	Block type	Layer configs
`encnet12`	5	64	`plain`	`[1, 1, 1, 1, 1]`
`encnet32`	5	64	`bottleneck`	`[2, 2, 2, 2, 2]`
`encnet47`	5	64	`bottleneck`	`[3, 3, 3, 3, 3]`
`encnet62`	5	64	`bottleneck`	`[4, 4, 4, 4, 4]`

Last update: March 14, 2021