File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -1087,8 +1087,9 @@ def _init_training_args(self) -> None:
10871087 kwargs ['neftune_noise_alpha' ] = self .neftune_noise_alpha
10881088
10891089 parameters = inspect .signature (Seq2SeqTrainingArguments .__init__ ).parameters
1090- if 'include_num_input_tokens_seen' in parameters :
1091- kwargs ['include_num_input_tokens_seen' ] = self .include_num_input_tokens_seen
1090+ for k in ['lr_scheduler_kwargs' , 'include_num_input_tokens_seen' ]:
1091+ if k in parameters :
1092+ kwargs [k ] = getattr (self , k )
10921093 if 'eval_strategy' in parameters :
10931094 kwargs ['eval_strategy' ] = self .evaluation_strategy
10941095 else :
@@ -1106,7 +1107,6 @@ def _init_training_args(self) -> None:
11061107 num_train_epochs = self .num_train_epochs ,
11071108 max_steps = self .max_steps ,
11081109 lr_scheduler_type = self .lr_scheduler_type ,
1109- lr_scheduler_kwargs = self .lr_scheduler_kwargs ,
11101110 warmup_ratio = self .warmup_ratio ,
11111111 warmup_steps = self .warmup_steps ,
11121112 logging_steps = self .logging_steps ,
You can’t perform that action at this time.
0 commit comments