From f75f941f1071fe048c74d4548eb33f3ff3f1720b Mon Sep 17 00:00:00 2001 From: Franck Dary <franck.dary@lis-lab.fr> Date: Fri, 3 Apr 2020 10:42:02 +0200 Subject: [PATCH] Print time in extract examples --- trainer/src/Trainer.cpp | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/trainer/src/Trainer.cpp b/trainer/src/Trainer.cpp index 621441c..0e61711 100644 --- a/trainer/src/Trainer.cpp +++ b/trainer/src/Trainer.cpp @@ -33,6 +33,8 @@ void Trainer::createDevDataset(SubConfig & config, bool debug) void Trainer::extractExamples(SubConfig & config, bool debug, std::vector<torch::Tensor> & contexts, std::vector<torch::Tensor> & classes) { + fmt::print(stderr, "[{}] Starting to extract examples\n", util::getTime()); + config.addPredicted(machine.getPredicted()); config.setState(machine.getStrategy().getInitialState()); @@ -88,6 +90,8 @@ void Trainer::extractExamples(SubConfig & config, bool debug, std::vector<torch: if (config.needsUpdate()) config.update(); } + + fmt::print(stderr, "[{}] Extracted {} examples\n", util::getTime(), util::int2HumanStr(classes.size())); } float Trainer::processDataset(DataLoader & loader, bool train, bool printAdvancement) -- GitLab