modules.resnet.ae*¶

Function · nn.Module

net = ae*(
    order=2, kernel_size=3, in_planes=1, out_planes=1
)

Instant presents of mdnc.module.resnet.AE*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_planes`	`int`	The channel number of the output data.

API	Net depth	Channel	Block type	Layer configs
`ae16`	3	64	`plain`	`[2, 1, 1]`
`ae32`	3	64	`bottleneck`	`[2, 2, 2]`
`ae44`	4	64	`bottleneck`	`[2, 2, 2, 2]`
`ae65`	4	64	`bottleneck`	`[3, 3, 3, 3]`
`ae83`	5	64	`bottleneck`	`[3, 3, 3, 3, 3]`

Last update: March 14, 2021