added keep_best and keep_after to test configs.

This commit is contained in:
gerazov
2021-03-08 02:56:36 +01:00
committed by Eren Gölge
parent 2db40457e8
commit 62147994d4
6 changed files with 187 additions and 175 deletions
+2
View File
@@ -111,6 +111,8 @@
"print_eval": false, // If True, it prints intermediate loss values in evalulation.
"save_step": 5000, // Number of training steps expected to save traninpg stats and checkpoints.
"checkpoint": true, // If true, it saves checkpoints per "save_step"
"keep_best": true, // If true, keeps all best_models after keep_after steps
"keep_after": 10000, // Global step after which to keep best models if keep_best is true
"tb_model_param_stats": false, // true, plots param stats per layer on tensorboard. Might be memory consuming, but good for debugging.:set n
"mixed_precision": false,