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

GPU implementation #24

Open

ps3-app opened this issue May 13, 2020 · 2 comments

ps3-app commented May 13, 2020

Is there any GPU implementation of main.py ?

gxwangupc commented May 17, 2020

Hi, just change namespace '--cuda' in arguments.py to
parser.add_argument('--cuda', default=True, help='If training is to be done on a GPU')

danishmsin commented May 17, 2020

Or else you could just pass the argument "--cuda" while running the script

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