diff --git a/configs/nat_base.json b/configs/nat_base.json index ab9d59e..271a880 100644 --- a/configs/nat_base.json +++ b/configs/nat_base.json @@ -7,7 +7,7 @@ "learning_rate": 2e-4, "betas": [0.8, 0.99], "eps": 1e-9, - "batch_size": 28, + "batch_size": 32, "fp16_run": true, "lr_decay": 0.999875, "segment_size": 8192,