SpUNetBase

class SpUNetBase(in_channels, num_classes=0, base_channels=32, context_channels=256, channels=(32, 64, 128, 256, 256, 128, 96, 96), layers=(2, 3, 4, 6, 2, 2, 2, 2), cls_mode=False, conditions=('ScanNet', 'S3DIS', 'Structured3D'), zero_init=True, norm_decouple=True, norm_adaptive=True, norm_affine=False)

Bases: Module

SpUNetBase.forward(input_dict)