Hi @ptrblck. I’ve posted a query, and haven’t received any reply, and I really need help with this. Your advice would really help me. Could you please have a look?

  
    
    
    PortraitNet model training - Loss not decreasing beyond initial point vision
  
  
    I’m trying to train a PortraitNet model in PyTorch for segmentation, with MobileNetv2 as my backend. I’m using SegmentAI as my dataset. My loss is not decreasing, it remains the same since the start. I don’t know why that’s happening. I’m pasting my model, and training code below, please help me! 
MODEL CODE:- 

class DepthwiseConv2d(nn.Module):
    def __init__(self, in_channels, out_channels, kernel_size=1, stride=1, padding=0, dilation=1, bias=False):
        super(DepthwiseConv2d, self).__in…