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

bug fix: init model when no checkpoint was loaded

Browse source
This commit is contained in:
xaedes 2023-08-28 01:48:21 +02:00
parent 4882ff0c59
commit 152cfaac36
No known key found for this signature in database
GPG key ID: 30030EDD817EA2B1
1 changed files with 3 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -2594,6 +2594,9 @@ int main(int argc, char ** argv) {
printf("%s: init model\n", __func__); printf("%s: init model\n", __func__);
bool existed = load_checkpoint_file(params.fn_checkpoint_in, &model, opt); bool existed = load_checkpoint_file(params.fn_checkpoint_in, &model, opt);
if (!existed) {
init_model(&model);
}
set_param_model(&model); set_param_model(&model);
opt->params = params.use_adam ? opt_params_adam : opt_params_lbfgs; opt->params = params.use_adam ? opt_params_adam : opt_params_lbfgs;