Update ds_config.json

43ef36f8 · zhaoying1 · bd2c2ab8 · 43ef36f8
Commit 43ef36f8 authored Dec 06, 2023 by zhaoying1
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

fine-tune/ds_config.json fine-tune/ds_config.json +1 -0

No files found.
--- a/fine-tune/ds_config.json
+++ b/fine-tune/ds_config.json
 {
    "train_micro_batch_size_per_gpu": "auto",
+    "train_batch_size": "auto",
    "zero_allow_untested_optimizer": true,
    "fp16": {
      "enabled": "auto",