search

yanwii / machine-translation

在tensor2tensor中使用自己的语料实现中英文翻译
23 stars 3 forks source link

GPU的要求 #4

Closed 6zhc closed 4 years ago

6zhc commented 4 years ago

我想问下，这个程序对GPU最低的要求是多少？在2080上能运行吗？

yanwii commented 4 years ago

没问题~但要适当的设置一下batch_size，不然会OOM