From 401e71e6d99c28bc1335dbe38ed888693b56da0c Mon Sep 17 00:00:00 2001
From: Franck Dary <franck.dary@lis-lab.fr>
Date: Sun, 10 May 2020 11:51:38 +0200
Subject: [PATCH] dynamic oracle disabled for segmenter
---
trainer/src/Trainer.cpp | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/trainer/src/Trainer.cpp b/trainer/src/Trainer.cpp
index 0136515..328b00c 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();
--
GitLab