Skip to content

Commit 5431a57

Browse files
authored
fix eval_dataset no (#2497)
1 parent 279b524 commit 5431a57

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

swift/llm/sft.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -355,11 +355,6 @@ def prepare_dataset(args, template: Template, msg: Optional[Dict[str, Any]] = No
355355
if use_torchacc():
356356
training_args.train_dataset_sample = train_dataset.shape[0] if train_dataset is not None else 0
357357

358-
if val_dataset is None:
359-
training_args.evaluation_strategy = IntervalStrategy.NO
360-
training_args.eval_strategy = IntervalStrategy.NO
361-
training_args.do_eval = False
362-
363358
tokenizer = template.tokenizer
364359
dataset_info = {}
365360
if args.packing:
@@ -415,6 +410,11 @@ def prepare_dataset(args, template: Template, msg: Optional[Dict[str, Any]] = No
415410
train_dataset = LazyLLMDataset(train_dataset, template.encode)
416411
if val_dataset is not None:
417412
val_dataset = LazyLLMDataset(val_dataset, template.encode)
413+
414+
if val_dataset is None:
415+
training_args.evaluation_strategy = IntervalStrategy.NO
416+
training_args.eval_strategy = IntervalStrategy.NO
417+
training_args.do_eval = False
418418
if isinstance(msg, dict):
419419
msg['dataset_info'] = dataset_info
420420
return train_dataset, val_dataset

0 commit comments

Comments
 (0)