From cfe3a83e2fb6558e354628ed78334e1c3b987071 Mon Sep 17 00:00:00 2001 From: Brandon Amos Date: Fri, 5 Feb 2016 20:08:13 -0500 Subject: [PATCH] websocket server: Use store_true for --cuda. --- demos/web/websocket-server.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/demos/web/websocket-server.py b/demos/web/websocket-server.py index ec36808..9e0f921 100755 --- 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,