New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

How to translate in one GPU while I train the model in 2 GPUs #8

Closed

duduscript opened this issue Aug 30, 2018 · 1 comment

duduscript commented Aug 30, 2018 •

edited

Loading

Hi,

I train transformer model in 2 GPUs and I translate WMT data in one GPU.

I find the computation is 100% in CPU rather than GPU, I need to compare between marian version and my old tensorflow version in one GPU.

How to run it in one GPU while I trained the model in 2 GPUs?

Author

duduscript commented Aug 30, 2018

It seems that works when I change the GPU setting both in command params and yml file

duduscript closed this as completed

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment