diff options
author | Taku Kudo <taku@google.com> | 2018-04-17 14:24:32 +0300 |
---|---|---|
committer | Taku Kudo <taku@google.com> | 2018-04-17 14:24:32 +0300 |
commit | 8a214d87725322f2412bb16d817a0fbcbf51ec0d (patch) | |
tree | 6291acbdb8fd07979847337055bc520e5b598380 /src/trainer_factory_test.cc | |
parent | 50be0cbbcb8c1a8293c6dbe0ac0b11c230137afb (diff) |
Moved the spec verifier and increases the sentencepiece_length param
Diffstat (limited to 'src/trainer_factory_test.cc')
-rw-r--r-- | src/trainer_factory_test.cc | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/src/trainer_factory_test.cc b/src/trainer_factory_test.cc index 672bfc8..8bd188e 100644 --- a/src/trainer_factory_test.cc +++ b/src/trainer_factory_test.cc @@ -21,6 +21,9 @@ TEST(TrainerFactoryTest, BasicTest) { TrainerSpec trainer_spec; NormalizerSpec normalizer_spec; + trainer_spec.set_model_prefix("model"); + trainer_spec.add_input("input"); + { trainer_spec.set_model_type(TrainerSpec::UNIGRAM); auto m = TrainerFactory::Create(trainer_spec, normalizer_spec); |