-
Notifications
You must be signed in to change notification settings - Fork 87
Open
Description
Thanks for your great work. In line 633 of gaussian_diffusion.py, terms["nll"] is calculated but not used. Whther it is a mistake, or whether it doesn't work.
terms["nll"] = self._token_discrete_loss(model_out_x_start, get_logits, input_ids_x, mask=input_ids_mask, truncate=True, t=t)
terms["loss"] = terms["mse"] + decoder_nll + tT_loss
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
No labels