Solved by albanD in post #4 
              
              
                I guess it’s like a resnet module: 
class ConcatMod(nn.Module):
    def __init__(self, args):
        # Use args properly to set conv params here depending on your application
        self.conv1 = nn.Conv3d(32, 64, kernel_size=3, stride=1)
        self.conv1_norm = nn.BatchNorm3d(64)
        self.co…