diff --git a/trainer/src/Trainer.cpp b/trainer/src/Trainer.cpp index 0136515a7852a95b873c8693c90167d053117465..328b00c141085d23dec63a6d23ea8e3f4eb09118 100644 --- a/trainer/src/Trainer.cpp +++ b/trainer/src/Trainer.cpp @@ -86,7 +86,7 @@ void Trainer::extractExamples(SubConfig & config, bool debug, std::filesystem::p goldTransition = machine.getTransitionSet().getBestAppliableTransition(config); - if (dynamicOracle and util::choiceWithProbability(0.8) and config.getState() != "tokenizer" and config.getState() != "parser") + if (dynamicOracle and util::choiceWithProbability(0.8) and config.getState() != "tokenizer" and config.getState() != "parser" and config.getState() != "segmenter") { auto neuralInput = torch::from_blob(context[0].data(), {(long)context[0].size()}, torch::kLong).clone().to(NeuralNetworkImpl::device); auto prediction = machine.getClassifier()->getNN()(neuralInput).squeeze();