add no_cuda args in extract_features
This commit is contained in:
@@ -208,6 +208,10 @@ def main():
|
|||||||
type=int,
|
type=int,
|
||||||
default=-1,
|
default=-1,
|
||||||
help = "local_rank for distributed training on gpus")
|
help = "local_rank for distributed training on gpus")
|
||||||
|
parser.add_argument("--no_cuda",
|
||||||
|
default=False,
|
||||||
|
action='store_true',
|
||||||
|
help="Whether not to use CUDA when available")
|
||||||
|
|
||||||
args = parser.parse_args()
|
args = parser.parse_args()
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user