From 256a9f80b2cb6bb85f9099b94956f809fbd66618 Mon Sep 17 00:00:00 2001
From: Franck Dary <franck.dary@lis-lab.fr>
Date: Mon, 29 Jul 2019 15:25:57 +0200
Subject: [PATCH] Cosmetic change to code
---
trainer/src/Trainer.cpp | 10 ++++++----
1 file changed, 6 insertions(+), 4 deletions(-)
diff --git a/trainer/src/Trainer.cpp b/trainer/src/Trainer.cpp
index 9f41dfc..76bf1bd 100644
--- a/trainer/src/Trainer.cpp
+++ b/trainer/src/Trainer.cpp
@@ -400,10 +400,6 @@ void Trainer::doStepTrain()
if (TI.lastActionWasPredicted[normalStateName])
{
- if (ProgramParameters::debug)
- {
- fprintf(stderr, "Updating neural network \'%s\'\n", tm.getCurrentClassifier()->name.c_str());
- }
if (newCost >= lastCost)
{
loss = tm.getCurrentClassifier()->trainOnExample(pendingFD[tm.getCurrentClassifier()->name], tm.getCurrentClassifier()->getActionIndex("EPSILON"));
@@ -411,8 +407,14 @@ void Trainer::doStepTrain()
else
{
loss = tm.getCurrentClassifier()->trainOnExample(pendingFD[tm.getCurrentClassifier()->name], tm.getCurrentClassifier()->getActionIndex(trainConfig.getCurrentStateHistory().top()));
+
+ if (ProgramParameters::debug)
+ fprintf(stderr, "Updating neural network \'%s\', gold=\'%s\'\n", tm.getCurrentClassifier()->name.c_str(), trainConfig.getCurrentStateHistory().top().c_str());
}
+ if (ProgramParameters::debug)
+ fprintf(stderr, "Updating neural network \'%s\'\n", tm.getCurrentClassifier()->name.c_str());
+
TI.addTrainLoss(tm.getCurrentClassifier()->name, loss);
}
--
GitLab