SpUNetBase#

class SpUNetBase(in_channels, num_classes, base_channels=32, channels=(32, 64, 128, 256, 256, 128, 96, 96), layers=(2, 3, 4, 6, 2, 2, 2, 2), bn_momentum=0.1)[source]#

Bases: Module

SpUNetBase.forward(data_dict)[source]#