[Fix] change save_every defaults to 1 (open-compass#592)

liuyaox · Nov 15, 2023 · 32bf116 · 32bf116
1 parent eb9aa85
commit 32bf116
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 8 deletions.
diff --git a/opencompass/openicl/icl_inferencer/icl_attack_inferencer.py b/opencompass/openicl/icl_inferencer/icl_attack_inferencer.py
@@ -42,7 +42,7 @@ class AttackInferencer(BaseInferencer):
         gen_field_replace_token (:obj:`str`, optional): Used to replace the
             generation field token when generating prompts.
         save_every (:obj:`int`, optional): Save intermediate results every
-            `save_every` epochs.
+            `save_every` iters. Defaults to 1.
         generation_kwargs (:obj:`Dict`, optional): Parameters for the
             :obj:`model.generate()` method.
     """
@@ -58,7 +58,7 @@ def __init__(
             gen_field_replace_token: Optional[str] = '',
             output_json_filepath: Optional[str] = './icl_inference_output',
             output_json_filename: Optional[str] = 'predictions',
-            save_every: Optional[int] = None,
+            save_every: Optional[int] = 1,
             dataset_cfg: Optional[List[int]] = None,
             **kwargs) -> None:
         super().__init__(

diff --git a/opencompass/openicl/icl_inferencer/icl_gen_inferencer.py b/opencompass/openicl/icl_inferencer/icl_gen_inferencer.py
@@ -36,7 +36,7 @@ class GenInferencer(BaseInferencer):
         gen_field_replace_token (:obj:`str`, optional): Used to replace the
             generation field token when generating prompts.
         save_every (:obj:`int`, optional): Save intermediate results every
-            `save_every` epochs.
+            `save_every` iters. Defaults to 1.
         generation_kwargs (:obj:`Dict`, optional): Parameters for the
             :obj:`model.generate()` method.
     """
@@ -50,7 +50,7 @@ def __init__(
             gen_field_replace_token: Optional[str] = '',
             output_json_filepath: Optional[str] = './icl_inference_output',
             output_json_filename: Optional[str] = 'predictions',
-            save_every: Optional[int] = None,
+            save_every: Optional[int] = 1,
             **kwargs) -> None:
         super().__init__(
             model=model,

diff --git a/opencompass/openicl/icl_inferencer/icl_sc_inferencer.py b/opencompass/openicl/icl_inferencer/icl_sc_inferencer.py
@@ -34,7 +34,7 @@ class SCInferencer(BaseInferencer):
         gen_field_replace_token (:obj:`str`, optional): Used to replace the
             generation field token when generating prompts.
         save_every (:obj:`int`, optional): Save intermediate results every
-            `save_every` epochs.
+            `save_every` iters. Defaults to 1.
         generation_kwargs (:obj:`Dict`, optional): Parameters for the
             :obj:`model.generate()` method.
         sc_size (:obj:`int`, optional): Sample size for Self-Consistency
@@ -51,7 +51,7 @@ def __init__(
             gen_field_replace_token: Optional[str] = '',
             output_json_filepath: Optional[str] = './icl_inference_output',
             output_json_filename: Optional[str] = 'predictions',
-            save_every: Optional[int] = None,
+            save_every: Optional[int] = 1,
             sc_size: Optional[int] = 1,
             infer_type: Optional[str] = '',
             generation_kwargs: dict = {},

diff --git a/opencompass/openicl/icl_inferencer/icl_tot_inferencer.py b/opencompass/openicl/icl_inferencer/icl_tot_inferencer.py
@@ -43,7 +43,7 @@ class ToTInferencer(GenInferencer):
         gen_field_replace_token (:obj:`str`, optional): Used to replace the
             generation field token when generating prompts.
         save_every (:obj:`int`, optional): Save intermediate results every
-            `save_every` epochs.
+            `save_every` iters. Defaults to 1.
         generation_kwargs (:obj:`Dict`, optional): Parameters for the
             :obj:`model.generate()` method.
         naive_run (:obj:`bool`): if True, run naive IO/CoT sampling instead of
@@ -74,7 +74,7 @@ def __init__(
             gen_field_replace_token: Optional[str] = '',
             output_json_filepath: Optional[str] = './icl_inference_output',
             output_json_filename: Optional[str] = 'predictions',
-            save_every: Optional[int] = None,
+            save_every: Optional[int] = 1,
             naive_run: bool = False,
             prompt_wrapper: dict = {},
             prompt_sample: str = 'standard',