search

reppy4620 / Dialog

A PyTorch Implementation of japanese chatbot using BERT and Transformer's decoder
MIT License
72 stars 29 forks source link

CrossEntropy #7

Closed reppy4620 closed 4 years ago

reppy4620 commented 4 years ago

Changed object function from KLDivLoss to CrossEntropy.