websocket server: Use store_true for --cuda.

2016-02-05 20:08:13 -05:00 · 2016-02-05 20:08:13 -05:00 · cfe3a83e2f
parent a5528889c4
commit cfe3a83e2f
1 changed files with 1 additions and 1 deletions
--- a/demos/web/websocket-server.py
+++ b/demos/web/websocket-server.py
@ -61,7 +61,7 @@ parser.add_argument('--networkModel', type=str, help="Path to Torch network mode
                    default=os.path.join(openfaceModelDir, 'nn4.small2.v1.t7'))
 parser.add_argument('--imgDim', type=int,
                    help="Default image dimension.", default=96)
-parser.add_argument('--cuda', type=bool, default=False)
+parser.add_argument('--cuda', action='store_true')
 parser.add_argument('--unknown', type=bool, default=False,
                    help='Try to predict unknown people')
 parser.add_argument('--port', type=int, default=9000,