From f2e0e3dadbaa7daa8456e067df083616bd45a108 Mon Sep 17 00:00:00 2001
From: Franck Dary <franck.dary@lis-lab.fr>
Date: Sun, 10 May 2020 21:20:58 +0200
Subject: [PATCH] Save everything if dynamic oracle is used
---
trainer/src/MacaonTrain.cpp | 5 +++++
1 file changed, 5 insertions(+)
diff --git a/trainer/src/MacaonTrain.cpp b/trainer/src/MacaonTrain.cpp
index 32468ad..efcab79 100644
--- a/trainer/src/MacaonTrain.cpp
+++ b/trainer/src/MacaonTrain.cpp
@@ -82,6 +82,7 @@ int MacaonTrain::main()
auto nbEpoch = variables["nbEpochs"].as<int>();
auto batchSize = variables["batchSize"].as<int>();
auto dynamicOracleInterval = variables["dynamicOracleInterval"].as<int>();
+ bool saveEverything = dynamicOracleInterval > 0;
auto rarityThreshold = variables["rarityThreshold"].as<float>();
bool debug = variables.count("debug") == 0 ? false : true;
bool printAdvancement = !debug && variables.count("silent") == 0 ? true : false;
@@ -199,8 +200,12 @@ int MacaonTrain::main()
devScoresStr.pop_back();
devScoreMean /= devScores.size();
bool saved = devScoreMean >= bestDevScore;
+
if (!computeDevScore)
saved = devScoreMean <= bestDevScore;
+
+ if (saveEverything)
+ saved = true;
if (saved)
{
bestDevScore = devScoreMean;
--
GitLab