-
Notifications
You must be signed in to change notification settings - Fork 150
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.
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
--use_gpu=False failed #13
Comments
Hi @sdlpkxd , plz refer to the updated README. thanks. |
When you use argparse to parse boolean command-line arguments, it doesn't work. In you code "parser.add_argument('--use_gpu', dest='use_gpu', type=bool, default=True, help='gpu flag')" in main.py, type=bool maybe Inappropriate. I can not change the work mode to CPU by "python main.py --phase test --use_gpu False". You can refer to this. thank you |
Yeah, it is a bug, waiting to be fixed. |
Thanks @sdlpkxd to fix it! |
I try to run your code without GPU using the flowing commands:
python main.py --phase=test --use_gpu=False
But it prints "GPU" and “[!] Load failed...", I think it should print "CPU". Can the type of bool be used in parser.add_argument()? I don't think so.
You can refer to https://stackoverflow.com/questions/15008758/parsing-boolean-values-with-argparse
The text was updated successfully, but these errors were encountered: