HCha/Stockfish
1
0
Fork 0
mirror of https://github.com/HChaZZY/Stockfish.git synced 2025-12-25 03:26:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

Change some learn parameter naming. Update docs.

Browse Source
This commit is contained in:
Tomasz Sobczyk
2021-03-14 14:12:35 +01:00
committed by nodchip
parent 591609c262
commit 5fdb48a7cb
2 changed files with 37 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
src/learn/learn.cpp
View File

@@ -1221,7 +1221,9 @@ namespace Learner
else if (option == "basedir") is >> base_dir;
// Mini batch size
else if (option == "batchsize") is >> params.mini_batch_size;
else if (option == "batchsize"
|| option == "epoch_size")
is >> params.mini_batch_size;
// learning rate
else if (option == "lr") is >> params.learning_rate;
@@ -1260,9 +1262,15 @@ namespace Learner
else if (option == "save_only_once") params.save_only_once = true;
else if (option == "no_shuffle") params.shuffle = false;
else if (option == "nn_batch_size") is >> nn_batch_size;
else if (option == "newbob_decay") is >> params.newbob_decay;
else if (option == "newbob_num_trials") is >> params.newbob_num_trials;
else if (option == "nn_batch_size"
|| option == "batch_size")
is >> nn_batch_size;
else if (option == "newbob_decay"
|| option == "lr_step")
is >> params.newbob_decay;
else if (option == "newbob_num_trials"
|| option == "max_consecutive_rejections")
is >> params.newbob_num_trials;
else if (option == "nn_options") is >> nn_options;
else if (option == "auto_lr_drop") is >> params.auto_lr_drop;