Opennmt-py: AttributeError: 'DataParallel' object has no attribute 'generator'

Created on 8 Oct 2017  路  4Comments  路  Source: OpenNMT/OpenNMT-py

Hey guys, it looks like the model having problem when passing more than one gpu id.
It crashes after trying to fetch the model's generator, as the DataParallel object doesn't have one (It only has the module object)
It looks like this breaks training not only in this specific line.

Any advice?
(In order to reproduce the problem I'm facing just pass -guid 0 1)

picture mataney

Most helpful comment

Yeah. Sorry, multigpu is actually not supported at the moment. We are waiting on some work from the pytorch team. Currently even if you get it to run, it is not much faster.

picture srush on 10 Oct 2017
👍3

All 4 comments

how to solve this question?

KelleyYin picture KelleyYin on 9 Oct 2017

the same problem...

XiangZhu0718 picture XiangZhu0718 on 10 Oct 2017

Could you post the trackback ? Is it the newest code?

JianyuZhan picture JianyuZhan on 10 Oct 2017

Yeah. Sorry, multigpu is actually not supported at the moment. We are waiting on some work from the pytorch team. Currently even if you get it to run, it is not much faster.

srush picture srush on 10 Oct 2017
👍3
Was this page helpful?
0 / 5 - 0 ratings

Related issues

1130310223 picture 1130310223  路  4Comments
mrpega picture mrpega  路  3Comments
KelleyYin picture KelleyYin  路  6Comments
mzeidhassan picture mzeidhassan  路  4Comments
paulkp picture paulkp  路  6Comments