Skip to content
This repository was archived by the owner on Nov 8, 2022. It is now read-only.

Commit 123655a

Browse files
author
Peter Izsak
authored
Fix token_tagging inference procedure
1 parent 5dcb99f commit 123655a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

nlp_architect/procedures/token_tagging.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -286,7 +286,7 @@ def do_kd_training(args):
286286
def do_inference(args):
287287
prepare_output_path(args.output_dir, args.overwrite_output_dir)
288288
device, n_gpus = setup_backend(args.no_cuda)
289-
args.batch_size = args.per_gpu_eval_batch_size * max(1, n_gpus)
289+
args.batch_size = args.b * max(1, n_gpus)
290290
inference_examples = process_inference_input(args.data_file)
291291
classifier = NeuralTagger.load_model(model_path=args.model_dir)
292292
classifier.to(device, n_gpus)

0 commit comments

Comments
 (0)