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

save checkpoint only when it was trained

Browse source
This commit is contained in:
xaedes 2023-05-22 16:56:28 +02:00
parent cc440bd438
commit 0651679302
No known key found for this signature in database
GPG key ID: 30030EDD817EA2B1
1 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
examples/baby-llama/baby-llama-text.cpp
View file

@ -2089,7 +2089,9 @@ int main(int argc, char ** argv) {
ggml_free(ctx0); ggml_free(ctx0);
} }
save_checkpoint(&model, opt, fn_chkpt_out); if (n_examples > 0) {
save_checkpoint(&model, opt, fn_chkpt_out);
}
{ {
int n_gen = 1024; int n_gen = 1024;