Can the code here be trained with multiple GPUs

atulkum / pointer_summarizer

pytorch implementation of "Get To The Point: Summarization with Pointer-Generator Networks"

Apache License 2.0

904 stars 243 forks source link

Closed gaozhiguang closed 3 years ago

gaozhiguang commented 3 years ago

Only one GPU is a little slow, how can i train with multiple GPUS?

atulkum commented 3 years ago

Currently it is not but it is good to try. You might need to first move the code (mostly data iterator) to new pytorch verison and python3.