websocket server: Use store_true for --cuda.

This commit is contained in:
Brandon Amos 2016-02-05 20:08:13 -05:00
parent a5528889c4
commit cfe3a83e2f
1 changed files with 1 additions and 1 deletions

View File

@ -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')) default=os.path.join(openfaceModelDir, 'nn4.small2.v1.t7'))
parser.add_argument('--imgDim', type=int, parser.add_argument('--imgDim', type=int,
help="Default image dimension.", default=96) 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, parser.add_argument('--unknown', type=bool, default=False,
help='Try to predict unknown people') help='Try to predict unknown people')
parser.add_argument('--port', type=int, default=9000, parser.add_argument('--port', type=int, default=9000,