From 918720792b8714ec7b02e2b3ef0e0bf6ec6a6ae4 Mon Sep 17 00:00:00 2001
From: Franck Dary <franck.dary@lis-lab.fr>
Date: Sun, 23 Feb 2020 19:53:37 +0100
Subject: [PATCH] Set neural network to eval mode in decode
---
decoder/src/Decoder.cpp | 3 +++
1 file changed, 3 insertions(+)
diff --git a/decoder/src/Decoder.cpp b/decoder/src/Decoder.cpp
index 2996b79..aa0b0c2 100644
--- a/decoder/src/Decoder.cpp
+++ b/decoder/src/Decoder.cpp
@@ -7,6 +7,7 @@ Decoder::Decoder(ReadingMachine & machine) : machine(machine)
void Decoder::decode(BaseConfig & config, std::size_t beamSize, bool debug)
{
+ machine.getClassifier()->getNN()->train(false);
config.addPredicted(machine.getPredicted());
try
@@ -63,6 +64,8 @@ void Decoder::decode(BaseConfig & config, std::size_t beamSize, bool debug)
if (debug)
fmt::print(stderr, "Forcing EOS transition\n");
}
+
+ machine.getClassifier()->getNN()->train(true);
}
float Decoder::getMetricScore(const std::string & metric, std::size_t scoreIndex) const
--
GitLab