From 31016256cffaf80cf4af8fb1b02f5565ea28478b Mon Sep 17 00:00:00 2001 From: Franck Dary <franck.dary@lis-lab.fr> Date: Sat, 9 Oct 2021 17:22:42 +0200 Subject: [PATCH] Disable parralelism --- decoder/src/MacaonDecode.cpp | 2 +- trainer/src/MacaonTrain.cpp | 2 +- trainer/src/Trainer.cpp | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/decoder/src/MacaonDecode.cpp b/decoder/src/MacaonDecode.cpp index 22e715f..074c6ad 100644 --- a/decoder/src/MacaonDecode.cpp +++ b/decoder/src/MacaonDecode.cpp @@ -129,7 +129,7 @@ int MacaonDecode::main() { NeuralNetworkImpl::setDevice(torch::kCPU); machine.to(NeuralNetworkImpl::getDevice()); - std::for_each(std::execution::par, configs.begin(), configs.end(), + std::for_each(std::execution::seq, configs.begin(), configs.end(), [&decoder, debug, printAdvancement, beamSize, beamThreshold](BaseConfig & config) { decoder.decode(config, beamSize, beamThreshold, debug, printAdvancement); diff --git a/trainer/src/MacaonTrain.cpp b/trainer/src/MacaonTrain.cpp index d760d1f..9846783 100644 --- a/trainer/src/MacaonTrain.cpp +++ b/trainer/src/MacaonTrain.cpp @@ -329,7 +329,7 @@ int MacaonTrain::main() { NeuralNetworkImpl::setDevice(torch::kCPU); machine.to(NeuralNetworkImpl::getDevice()); - std::for_each(std::execution::par, devConfigs.begin(), devConfigs.end(), + std::for_each(std::execution::seq, devConfigs.begin(), devConfigs.end(), [&decoder, debug, printAdvancement](BaseConfig & devConfig) { decoder.decode(devConfig, 1, 0.0, debug, printAdvancement); diff --git a/trainer/src/Trainer.cpp b/trainer/src/Trainer.cpp index aaef532..da9fb73 100644 --- a/trainer/src/Trainer.cpp +++ b/trainer/src/Trainer.cpp @@ -55,7 +55,7 @@ void Trainer::extractExamples(std::vector<SubConfig> & configs, bool debug, std: NeuralNetworkImpl::setDevice(torch::kCPU); machine.to(NeuralNetworkImpl::getDevice()); - std::for_each(std::execution::par, configs.begin(), configs.end(), + std::for_each(std::execution::seq, configs.begin(), configs.end(), [this, maxNbExamplesPerFile, &examplesPerState, &totalNbExamples, debug, memcheck, dynamicOracle, explorationThreshold, dir, epoch, &examplesMutex](SubConfig & config) { config.addPredicted(machine.getPredicted()); -- GitLab