From 6b37a8c65660cf2b1c563db6c0f86ff55f80a7dc Mon Sep 17 00:00:00 2001 From: Max Marion Date: Mon, 12 Feb 2024 19:44:28 +0000 Subject: [PATCH] change defaults --- llmfoundry/utils/builders.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/llmfoundry/utils/builders.py b/llmfoundry/utils/builders.py index 391b792a1c..103110615d 100644 --- a/llmfoundry/utils/builders.py +++ b/llmfoundry/utils/builders.py @@ -53,7 +53,7 @@ def build_evaluators( device_eval_batch_size: int, icl_seq_len: int, icl_subset_num_batches: Optional[int], - fewshot_random_seed: Optional[int] = None, + fewshot_random_seed: Optional[int] = 1234, ) -> Tuple[List[Evaluator], List[str], Optional[EvalGauntlet]]: evaluators = [] @@ -130,7 +130,7 @@ def build_icl_data_and_gauntlet( tokenizer: PreTrainedTokenizerBase, device_eval_batch_size: int, icl_seq_len: int, - fewshot_random_seed: Optional[int], + fewshot_random_seed: Optional[int] = 1234, icl_subset_num_batches: Optional[int] = None ) -> Tuple[List[Evaluator], List[str], Optional[EvalGauntlet]]: icl_evaluators, logger_keys = build_icl_evaluators( @@ -446,7 +446,7 @@ def build_icl_evaluators( default_max_seq_len: int, default_batch_size: int, destination_dir: Optional[str] = None, - fewshot_random_seed: Optional[int] = None, + fewshot_random_seed: Optional[int] = 1234, icl_subset_num_batches: Optional[int] = None, ) -> Tuple[List[Evaluator], List[str]]: if destination_dir is None: