1、默认使用cpu运行，若想使用gpu，将代码第33行的DEVICE = ‘cpu’ 改为DEVICE = ‘cuda’即可
2、python transformer_nmt.py执行
forked from seanzhang-zhichen/-transformer-english2chinese-
-
Notifications
You must be signed in to change notification settings - Fork 0
qianmin/transformer-en2cn
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
pytorch
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published
Languages
- Python 100.0%