/pytorch-transformer

pytorch复现transformer

Primary LanguagePython

pytorch-transformer

pytorch复现transformer

数据集: 德语翻译英语

运行效果

$ python evaluation.py
Zwei junge weiße Männer sind im Freien in der Nähe vieler Büsche. -> Two young , White males are outside near many bushes .
Mehrere Männer mit Schutzhelmen bedienen ein Antriebsradsystem. -> Several men in hard hats are operating a giant pulley system .
Ein kleines Mädchen klettert in ein Spielhaus aus Holz. -> A little girl climbing into a wooden playhouse .
Ein Mann in einem blauen Hemd steht auf einer Leiter und putzt ein Fenster. -> A man in a blue shirt is standing on a ladder cleaning a window .
Zwei Männer stehen am Herd und bereiten Essen zu. -> Two men are at the stove preparing food .
Ein Mann in grün hält eine Gitarre, während der andere Mann sein Hemd ansieht. -> A man in green holds a guitar while the other man observes his shirt .
Ein Mann lächelt einen ausgestopften Löwen an. -> A man is smiling at a stuffed lion
Ein schickes Mädchen spricht mit dem Handy während sie langsam die Straße entlangschwebt. -> A trendy girl talking on her cellphone while gliding slowly down the street .

说明

ref: 理论
checkpoints: 训练好的model
train.py:训练
evaluation.py:预测

了解更多

代码和资源均免费公开,但如果你觉得很难看懂的话,那你大概率应该加入我的私教小课堂:

我会在鱼塘里做2件事情:
1,回答大家关于本项目的问题
2,定期组织直播课讲解本项目

ai小鱼塘