diff --git a/trainer/src/Trainer.cpp b/trainer/src/Trainer.cpp index 3b43a98e27c56c6273d647835e934beb48419450..eb76934d0b58ddc14f8f392776eeef0b07de5c1b 100644 --- a/trainer/src/Trainer.cpp +++ b/trainer/src/Trainer.cpp @@ -88,7 +88,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" and config.getState() != "segmenter") + if (dynamicOracle and util::choiceWithProbability(0.8) and config.getState() != "tokenizer" 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();