Hey,
I am using the following implementation to apply gumble softmax on my output layer. What loss function should I use with gumble softmax.

  
      github.com/pytorch/pytorch
  
  
      
    
  


  add gumbel_softmax, based on Eric Jang's implementation 55



  by hughperkins
  on 04:15PM - 28 Oct 17



  3 commits
  changed 2 files
  with 120 additions
  and 0 deletions.


  
  
    
    
  
  


I have removed the softmax after applying gumble noise and kept my loss function as CrossEntropyLoss. Is that fine?
Regards