Predictor args

kfnxu · Mar 10, 2018 · c405826 · c405826
1 parent 4180963
commit c405826
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/scripts/reader/predict.py b/scripts/reader/predict.py
@@ -60,10 +60,10 @@
     logger.info('Running on CPU only.')
 
 predictor = Predictor(
-    args.model,
-    args.tokenizer,
-    args.embedding_file,
-    args.num_workers,
+    model=args.model,
+    tokenizer=args.tokenizer,
+    embedding_file=args.embedding_file,
+    num_workers=args.num_workers,
 )
 if args.cuda:
     predictor.cuda()