diff --git a/trainer/src/Trainer.cpp b/trainer/src/Trainer.cpp index 30c535268752e6ad91f0c15ee88a726155563c60..d05667fafff86114d9a16fc17ec93a153c098c6d 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) {