Closed yexxx closed 10 months ago
the d in vat_loss will not update when delta_kl.backward(), need to add this line to update d.
the d in vat_loss will not update when delta_kl.backward(), need to add this line to update d.