Add eval_batch_size for evaluation

2023-05-06 22:21:24 +09:00
parent c0f50d9c61
commit 0e74b6402e
2 changed files with 2 additions and 0 deletions
--- a/README.md
+++ b/README.md
@@ -85,6 +85,7 @@ output_dir: ./completed-model
 # training hyperparameters
 batch_size: 8
 micro_batch_size: 2
+eval_batch_size: 2
 num_epochs: 3
 warmup_steps: 100
 learning_rate: 0.00003