欢迎关注我的CSDN:https://blog.csdn.net/caroline_wendy
本文地址:https://blog.csdn.net/caroline_wendy/article/details/128222398
NLP seq2seq代码编写技巧
collate_fn:pad_sequence、 label=target[:,1:]、 decoder_input=target[:, :-1] torch.nn.utils.rnn.pad_sequencecollate_fn输入是多个训练对,batch_size个