Can you sharee the logic why Batchnorm is still used in the first spconv layer and in the (un)Pool layers?
Is there anything against replacing this with a LayerNorm, GroupNorm?
PS: If I understand it right, for the SerializedPooling, the BatchNorm would build the statistics over all points in all batches.