We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent f76ae69 commit f9e2ea3Copy full SHA for f9e2ea3
1 file changed
src/maxtext/trainers/post_train/rl/train_rl.py
@@ -337,7 +337,7 @@ def prepare_datasets(trainer_config, model_tokenizer):
337
"nvidia/OpenMathReasoning",
338
"open-r1/OpenR1-Math-220k",
339
"bethgelab/CuratedThoughts",
340
- ] and (not eval_dataset_name or eval_dataset_name == trainer_config.dataset_name):
+ ] and eval_dataset_name == trainer_config.dataset_name:
341
import datasets # pylint: disable=import-outside-toplevel
342
343
splits = prepare_train_and_eval_dataset(trainer_config)
0 commit comments