|
|
@ -45,12 +45,12 @@ parser.add_argument(
|
|
|
|
help="Use gpu or not. (default: %(default)s)")
|
|
|
|
help="Use gpu or not. (default: %(default)s)")
|
|
|
|
parser.add_argument(
|
|
|
|
parser.add_argument(
|
|
|
|
"--num_threads_data",
|
|
|
|
"--num_threads_data",
|
|
|
|
default=multiprocessing.cpu_count(),
|
|
|
|
default=multiprocessing.cpu_count() // 2,
|
|
|
|
type=int,
|
|
|
|
type=int,
|
|
|
|
help="Number of cpu threads for preprocessing data. (default: %(default)s)")
|
|
|
|
help="Number of cpu threads for preprocessing data. (default: %(default)s)")
|
|
|
|
parser.add_argument(
|
|
|
|
parser.add_argument(
|
|
|
|
"--num_processes_beam_search",
|
|
|
|
"--num_processes_beam_search",
|
|
|
|
default=multiprocessing.cpu_count(),
|
|
|
|
default=multiprocessing.cpu_count() // 2,
|
|
|
|
type=int,
|
|
|
|
type=int,
|
|
|
|
help="Number of cpu processes for beam search. (default: %(default)s)")
|
|
|
|
help="Number of cpu processes for beam search. (default: %(default)s)")
|
|
|
|
parser.add_argument(
|
|
|
|
parser.add_argument(
|
|
|
|