As it might be helpful to others, I posted an example of usage of nn.CrossEntropyLoss for image segmentation here :
  
      stackoverflow.com
  
  
      
    
  

  Is there a version of sparse categorical cross entropy in pytorch? 132



  pytorch, loss-function, cnn



  
  answered by
  
    Ismael EL ATIFI
  
  on 09:37PM - 26 Mar 21 UTC