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