From 695d71f0e5c8952b783d01f200acc00ebd123b7c Mon Sep 17 00:00:00 2001
From: Franck Dary <franck.dary@lis-lab.fr>
Date: Fri, 25 Oct 2019 15:22:24 +0200
Subject: [PATCH] Fixed display of advancement not going to 100 percent
---
trainer/src/Trainer.cpp | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/trainer/src/Trainer.cpp b/trainer/src/Trainer.cpp
index 30c5352..d05667f 100644
--- a/trainer/src/Trainer.cpp
+++ b/trainer/src/Trainer.cpp
@@ -16,7 +16,7 @@ Trainer::Trainer(TransitionMachine & tm, BD & bd, Config & config)
nbSteps = 0;
nbActions = 0;
- nbActionsCutoff = 2*ProgramParameters::batchSize;
+ nbActionsCutoff = 8*ProgramParameters::batchSize;
currentSpeed = 0.0;
pastTime = std::chrono::high_resolution_clock::now();
}
@@ -96,7 +96,7 @@ void Trainer::computeScoreOnDev()
// Print current iter advancement in percentage
if (ProgramParameters::interactive)
{
- int totalSize = ProgramParameters::devTapeSize;
+ int totalSize = devConfig->getTape("FORM").size();
int steps = devConfig->getHead();
if (devConfig->rawInputHeadIndex > 0)
{
@@ -256,7 +256,7 @@ void Trainer::doStepTrain()
// Print current iter advancement in percentage
if (ProgramParameters::interactive)
{
- int totalSize = ProgramParameters::iterationSize == -1 ? ProgramParameters::tapeSize : ProgramParameters::iterationSize;
+ int totalSize = ProgramParameters::iterationSize == -1 ? trainConfig.getTape("FORM").size() : ProgramParameters::iterationSize;
int steps = ProgramParameters::iterationSize == -1 ? trainConfig.getHead() : nbSteps;
if (trainConfig.rawInputHeadIndex > 0)
{
--
GitLab