From e843d6e71cea22eaa9a4288138ef02bc8cc50e7d Mon Sep 17 00:00:00 2001 From: xaedes Date: Sun, 2 Jul 2023 21:38:52 +0200 Subject: [PATCH] measure and print total training time --- .../train-text-from-scratch/train-text-from-scratch.cpp | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/examples/train-text-from-scratch/train-text-from-scratch.cpp b/examples/train-text-from-scratch/train-text-from-scratch.cpp index de71dc996..0f330fd4a 100644 --- a/examples/train-text-from-scratch/train-text-from-scratch.cpp +++ b/examples/train-text-from-scratch/train-text-from-scratch.cpp @@ -3975,6 +3975,8 @@ int main(int argc, char ** argv) { printf("%s: begin training\n", __func__); + int64_t t0 = ggml_time_ms(); + for (int ex = 0; ex < params.n_examples; ++ex) { if (ex*n_batch >= (int) train_samples.size()) { shuffle_ints(train_samples.data(), train_samples.data() + train_samples.size()); @@ -4112,6 +4114,11 @@ int main(int argc, char ** argv) { ggml_free(ctx0); } + int64_t t1 = ggml_time_ms(); + int64_t d = t1-t0; + double dd = (double) d * 1e-3; + printf("%s: total training time=%f seconds\n", __func__, dd); + if (params.n_examples > 0) { save_checkpoint(&model, opt, params.fn_checkpoint_out); }