Update trainer_interface.cc

google · Jan 29, 2019 · 3d597a5 · 3d597a5
1 parent 552937e
commit 3d597a5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/trainer_interface.cc b/src/trainer_interface.cc
@@ -422,7 +422,7 @@ util::Status TrainerInterface::LoadSentences() {
   // +3 for meta pieces.
   if (trainer_spec_.model_type() != TrainerSpec::WORD &&
       trainer_spec_.model_type() != TrainerSpec::CHAR) {
-    CHECK_LT_OR_RETURN(
+    CHECK_LE_OR_RETURN(
         static_cast<int>(required_chars_.size() + meta_pieces_.size()),
         trainer_spec_.vocab_size())
         << "Vocabulary size is smaller than required_chars. "