hemingkx / ChineseNMT

ChineseNMT: Translate English to Chinese with PyTorch Implementation of Transformer
439 stars 84 forks source link

epoch超长时间是为什么呢? #4

Closed tan-yishu closed 3 years ago

tan-yishu commented 3 years ago

image 2GPU,8CPU。。一个epoch要一个多小时。。555

hemingkx commented 3 years ago

训练时间在readme和知乎文章中已经说明了哦,一个小时是正常的。训练时间长是因为本项目使用了未经优化的harvard-transformer,仅供大家理解transformer和NMT项目使用。如果想缩短训练时间或者开发更为实用的翻译项目,建议使用优化程度更高的fairseq~

tan-yishu commented 3 years ago

训练时间在readme和知乎文章中已经说明了哦,一个小时是正常的。训练时间长是因为本项目使用了未经优化的harvard-transformer,仅供大家理解transformer和NMT项目使用。如果想缩短训练时间或者开发更为实用的翻译项目,建议使用优化程度更高的fairseq~

谢谢~