abisee / pointer-generator

Code for the ACL 2017 paper "Get To The Point: Summarization with Pointer-Generator Networks"
Other
2.18k stars 812 forks source link

Multi-GPU support? #58

Open wjpjet opened 7 years ago

wjpjet commented 7 years ago

As this is based on TextSum code, is there an option for multi-GPU training?

abisee commented 7 years ago

We do not plan to add that to this repo, but it should be fairly straightforward to copy that functionality from the TextSum code.

v3nm commented 6 years ago

@wjpjet have you tried running it on multi gpu , i tried it on 3 gpu system but it is using only one gpu. @abisee can you help regarding it.

Shujian2015 commented 6 years ago

@v3nm There are some discussions: https://github.com/google/seq2seq/issues/44