pytorch官方教程中有一個(gè)Chatbot教程,就是利用seq2seq和注意力機(jī)制實(shí)現(xiàn)的,感覺(jué)和機(jī)器翻譯沒(méi)什么不同啊,如果對(duì)話中一句話有下一句,那么就把這一對(duì)句子加入模型進(jìn)行訓(xùn)練。其中在訓(xùn)練階段,損失函數(shù)通常需要進(jìn)行mask操作,因?yàn)橐粋€(gè)batch中句子的長(zhǎng)度通常是不一樣的,一個(gè)batch中不足長(zhǎng)度的位置需要進(jìn)行填充(pad)補(bǔ)0,最后生...
www.dbjr.com.cn/article/1806...htm 2025-6-6