diff --git a/trainer/src/Trainer.cpp b/trainer/src/Trainer.cpp index 621441c23bb84833cd60d66ffc0142c5c03935af..0e61711696916342b2224eeca9bac5db58d6b298 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)