Project Beatrice commited on
Commit
5f0ed96
1 Parent(s): 7ad9cbe

Perform validation at step 5000

Browse files
Files changed (1) hide show
  1. beatrice_trainer/__main__.py +2 -0
beatrice_trainer/__main__.py CHANGED
@@ -2724,6 +2724,7 @@ if __name__ == "__main__":
2724
  # === 4. 検証 ===
2725
  if (iteration + 1) % 50000 == 0 or iteration + 1 in {
2726
  1,
 
2727
  10000,
2728
  30000,
2729
  h.n_steps,
@@ -2819,6 +2820,7 @@ if __name__ == "__main__":
2819
  # === 5. 保存 ===
2820
  if (iteration + 1) % 50000 == 0 or iteration + 1 in {
2821
  1,
 
2822
  10000,
2823
  30000,
2824
  h.n_steps,
 
2724
  # === 4. 検証 ===
2725
  if (iteration + 1) % 50000 == 0 or iteration + 1 in {
2726
  1,
2727
+ 5000,
2728
  10000,
2729
  30000,
2730
  h.n_steps,
 
2820
  # === 5. 保存 ===
2821
  if (iteration + 1) % 50000 == 0 or iteration + 1 in {
2822
  1,
2823
+ 5000,
2824
  10000,
2825
  30000,
2826
  h.n_steps,