Opennmt-py: Anybody use multi-gpu for this systerm ?

Created on 12 Feb 2018  路  6Comments  路  Source: OpenNMT/OpenNMT-py

This systerm is very faster in only one gpu . However , we want to train the corpus of WMT which will cost too much times. Futhermore , we seek to spend less time for waiting experimental result .

contributions welcome feature

Most helpful comment

if you're in a hurry, try this fork:
https://github.com/Ubiqus/OpenNMT-py

All 6 comments

It's the next thing we plan on working on. We were waiting for some pytorch features.

maybe you can utilize nn.DataParallel function in pytorch v0.3, usage like this torch.nn.DataParallel(net, device_ids=[0, 1, 2]).cuda()

Yeah, we had that originally, but it didn't give speed improvements. We are coming back to it now.

Multi GPU still not working

if you're in a hurry, try this fork:
https://github.com/Ubiqus/OpenNMT-py

@vince62s : thank you for the link.

Was this page helpful?
0 / 5 - 0 ratings

Related issues

mrpega picture mrpega  路  3Comments

andreaferretti picture andreaferretti  路  5Comments

mataney picture mataney  路  4Comments

howardyclo picture howardyclo  路  6Comments

AyaNsar picture AyaNsar  路  4Comments