From 3863f7648e5d8edb571ac592f3ac4f5f0695275a Mon Sep 17 00:00:00 2001
From: Taku Kudo <taku@google.com>
Date: Sun, 30 Apr 2023 17:37:15 +0000
Subject: [PATCH] increases the max number of threads

---
 src/trainer_interface.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/trainer_interface.cc b/src/trainer_interface.cc
index 8e4bccc1..968f7b9c 100644
--- a/src/trainer_interface.cc
+++ b/src/trainer_interface.cc
@@ -67,7 +67,7 @@ util::Status VerifySpec(const TrainerSpec &trainer_spec) {
   CHECK_RANGE(trainer_spec.character_coverage(), 0.98, 1.0);
   CHECK_RANGE(trainer_spec.max_sentencepiece_length(), 1, 512);
   CHECK_RANGE(trainer_spec.num_sub_iterations(), 1, 10);
-  CHECK_RANGE(trainer_spec.num_threads(), 1, 128);
+  CHECK_RANGE(trainer_spec.num_threads(), 1, 1024);
   CHECK_RANGE(trainer_spec.self_test_sample_size(), 0, 1000);
   CHECK_RANGE(trainer_spec.shrinking_factor(), 0.5, 0.95);
   CHECK_RANGE(trainer_spec.max_sentence_length(), 10, 1073741824);