websocket server: Use store_true for --cuda.
This commit is contained in:
parent
a5528889c4
commit
cfe3a83e2f
|
@ -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,
|
||||||
|
|
Loading…
Reference in New Issue