diff --git a/llmfoundry/utils/config_utils.py b/llmfoundry/utils/config_utils.py index 6b63cc669f..a2a8767b63 100644 --- a/llmfoundry/utils/config_utils.py +++ b/llmfoundry/utils/config_utils.py @@ -76,6 +76,9 @@ class EvalConfig: run_name: Optional[str] = None metadata: Optional[Dict[str, str]] = None + # Dataloader + global_train_batch_size: Optional[int] = None + # Distributed parameters dist_timeout: Union[float, int] = 600.0 fsdp_config: Optional[Dict[str, Any]] = None