vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

measure and print total training time

Browse source
This commit is contained in:
xaedes 2023-07-02 21:38:52 +02:00
parent ff759d957c
commit e843d6e71c
No known key found for this signature in database
GPG key ID: 30030EDD817EA2B1
1 changed files with 7 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

7
examples/train-text-from-scratch/train-text-from-scratch.cpp
View file

@ -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);
}