Skip to content
Snippets Groups Projects
Commit 695d71f0 authored by Franck Dary's avatar Franck Dary
Browse files

Fixed display of advancement not going to 100 percent

parent be99c375
No related branches found
No related tags found
No related merge requests found
......@@ -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)
{
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment