New-Pytorch-Chinese
New-Pytorch-Chinese copied to clipboard
bug修改共享
感谢博主的分享 我用的pytorch版本是1.7.1,运行时遇到一个小问题 beam_search文件51行需要修改 beams_order = (best_scores_id.squeeze( 1) / n_extended_vocab).type(T.long) #(beam,); sorted
感谢您的分享,我修改一下。