MinkowskiEngine.modules 包

子模块

MinkowskiEngine.modules.resnet_block 模块

class MinkowskiEngine.modules.resnet_block.BasicBlock(inplanes, planes, stride=1, dilation=1, downsample=None, bn_momentum=0.1, dimension=- 1)

基类: torch.nn.modules.module.Module

expansion = 1

forward(x)

定义每次调用时执行的计算。

应由所有子类重写。

注意

虽然前向传播的实现需要在该函数内定义，但之后应调用 Module 实例而非该函数本身，因为前者负责运行注册的钩子，而后者则默默忽略它们。

training: bool

class MinkowskiEngine.modules.resnet_block.Bottleneck(inplanes, planes, stride=1, dilation=1, downsample=None, bn_momentum=0.1, dimension=- 1)

基类: torch.nn.modules.module.Module

expansion = 4

forward(x)

定义每次调用时执行的计算。

应由所有子类重写。

注意

虽然前向传播的实现需要在该函数内定义，但之后应调用 Module 实例而非该函数本身，因为前者负责运行注册的钩子，而后者则默默忽略它们。

training: bool

MinkowskiEngine.modules.senet_block 模块

class MinkowskiEngine.modules.senet_block.SEBasicBlock(inplanes, planes, stride=1, dilation=1, downsample=None, reduction=16, D=- 1)

基类: MinkowskiEngine.modules.resnet_block.BasicBlock

forward(x)

定义每次调用时执行的计算。

应由所有子类重写。

注意

虽然前向传播的实现需要在该函数内定义，但之后应调用 Module 实例而非该函数本身，因为前者负责运行注册的钩子，而后者则默默忽略它们。

training: bool

class MinkowskiEngine.modules.senet_block.SEBottleneck(inplanes, planes, stride=1, dilation=1, downsample=None, D=3, reduction=16)

基类: MinkowskiEngine.modules.resnet_block.Bottleneck

forward(x)

定义每次调用时执行的计算。

应由所有子类重写。

注意

虽然前向传播的实现需要在该函数内定义，但之后应调用 Module 实例而非该函数本身，因为前者负责运行注册的钩子，而后者则默默忽略它们。

training: bool

class MinkowskiEngine.modules.senet_block.SELayer(channel, reduction=16, D=- 1)

基类: torch.nn.modules.module.Module

forward(x)

定义每次调用时执行的计算。

应由所有子类重写。

注意

虽然前向传播的实现需要在该函数内定义，但之后应调用 Module 实例而非该函数本身，因为前者负责运行注册的钩子，而后者则默默忽略它们。

training: bool

模块内容