EntropyLoss is error

[BridgetteSong]

code defines as following:
in the loss_fuction.py

def entropy(self, logits, targets):
      log_q = F.log_softmax(logits, dim=-1)
      return -torch.mean(torch.sum(targets * log_q, dim=-1))

in the module.py
cat_prob = F.softmax(self.categorical_layer(z), dim=-1)

1. so in the entropy function, the first is logits of model output, the second is target, but you use entropy function with wrong orders int the get_encoder_loss
cat_lambda*(-self.entropy(cat_target, prob_)
2. entropy loss is already NLL loss, so it should not have a negative weight of -1, int the get_encoder_loss
cat_lambda*(-self.entropy(cat_target, prob_)
3. because you use F.log_softmax, so the logits should be log_probs, but in your module.py you use a softmax activation, so the log_q should be torch.log(logits)

so the right loss is following:

def entropy(self, logits, targets):
      log_q = torch.log(logits)
      return -torch.mean(torch.sum(targets * log_q, dim=-1))

def get_encoder_loss(self, id_, prob_, classes_, cat_lambda, kl_lambda, encoder_type):
        cat_target = self.indices_to_one_hot(id_, classes_)

        if (encoder_type == 'gst' or encoder_type == 'x-vector') and cat_lambda != 0.0:
            loss = cat_lambda*(self.entropy(prob_, cat_target) - np.log(0.1))
        elif (encoder_type == 'vae' or encoder_type == 'gst_vae') and (cat_lambda != 0.0 or kl_lambda !=0.0):
            loss = cat_lambda*(self.entropy(prob_[2], cat_target) - np.log(0.1)) + kl_lambda*self.KL_loss(prob_[0], prob_[1])
        elif encoder_type == 'gmvae' and (cat_lambda != 0.0 or kl_lambda !=0.0) :
            loss = self.gaussian_loss(prob_[0], prob_[1], prob_[2], prob_[3], prob_[4])*kl_lambda + (self.entropy(prob_[5], cat_target) - np.log(0.1))*cat_lambda
        else:
            loss = 0.0

        return loss