From 152cfaac3643c976321695eef97a54f1d837b81b Mon Sep 17 00:00:00 2001 From: xaedes Date: Mon, 28 Aug 2023 01:48:21 +0200 Subject: [PATCH] bug fix: init model when no checkpoint was loaded --- examples/train-text-from-scratch/train-text-from-scratch.cpp | 3 +++ 1 file changed, 3 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 5d5e48633..18683f18f 100644 --- a/examples/train-text-from-scratch/train-text-from-scratch.cpp +++ b/examples/train-text-from-scratch/train-text-from-scratch.cpp @@ -2594,6 +2594,9 @@ int main(int argc, char ** argv) { printf("%s: init model\n", __func__); bool existed = load_checkpoint_file(params.fn_checkpoint_in, &model, opt); + if (!existed) { + init_model(&model); + } set_param_model(&model); opt->params = params.use_adam ? opt_params_adam : opt_params_lbfgs;