MCPcopy Create free account
hub / github.com/Tiiny-AI/PowerInfer / get_default_train_params

Function get_default_train_params

examples/finetune/finetune.cpp:1233–1280  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

1231};
1232
1233static struct train_params get_default_train_params() {
1234 struct train_params params;
1235 params.common = get_default_train_params_common();
1236 params.fn_model_base = "";
1237 params.fn_lora_out = "ggml-lora-ITERATION-f32.gguf";
1238
1239 params.only_write_lora = false;
1240
1241 params.f_norm_rms_eps = 1e-5f;
1242 params.rope_freq_base = 10000.0f;
1243 params.rope_freq_scale = 1.0f;
1244
1245 params.custom_f_norm_rms_eps = false;
1246 params.custom_rope_freq_base = false;
1247 params.custom_rope_freq_scale = false;
1248
1249 params.lora_r = 4;
1250 params.lora_alpha = 4;
1251 params.custom_lora_alpha = false;
1252
1253 params.n_rank_attention_norm = 1;
1254 params.n_rank_wq = 4;
1255 params.n_rank_wk = 4;
1256 params.n_rank_wv = 4;
1257 params.n_rank_wo = 4;
1258 params.n_rank_ffn_norm = 1;
1259 params.n_rank_w1 = 4;
1260 params.n_rank_w2 = 4;
1261 params.n_rank_w3 = 4;
1262 params.n_rank_tok_embeddings = 4;
1263 params.n_rank_norm = 1;
1264 params.n_rank_output = 4;
1265
1266 params.custom_n_rank_attention_norm = false;
1267 params.custom_n_rank_wq = false;
1268 params.custom_n_rank_wk = false;
1269 params.custom_n_rank_wv = false;
1270 params.custom_n_rank_wo = false;
1271 params.custom_n_rank_ffn_norm = false;
1272 params.custom_n_rank_w1 = false;
1273 params.custom_n_rank_w2 = false;
1274 params.custom_n_rank_w3 = false;
1275 params.custom_n_rank_tok_embeddings = false;
1276 params.custom_n_rank_norm = false;
1277 params.custom_n_rank_output = false;
1278
1279 return params;
1280}
1281
1282static void train_print_usage(int argc, char ** argv, const struct train_params * params) {
1283 fprintf(stderr, "usage: %s [options]\n", argv[0]);

Callers 2

train_params_parseFunction · 0.70
mainFunction · 0.70

Calls 1

Tested by

no test coverage detected