te of kernel
            norm_layer (callable, optional): Module specifying the normalization layer to use. Default: None
        r