update to conform to json format

This commit is contained in:
Tong Li 2025-05-28 13:13:21 +08:00 committed by xysheng-colossal
parent 2dd59c0ddd
commit 2783ddecf5
2 changed files with 2 additions and 2 deletions

View File

@ -178,7 +178,7 @@ Each data sample in the training or evaluation `.jsonl` file should follow this
| ---------------- | --------------------------------------- | ----------------- |
| `--model` | Model path or identifier | `/path/to/model` |
| `--dataset` | Path to training `.jsonl` | `/path/to/train_data.jsonl` |
| `--eval-dataset` | JSON of task\:eval\_dataset\_path pairs | `{'eval_1':'/path/to/eval_1.jsonl'}` |
| `--eval-dataset` | JSON of task\:eval\_dataset\_path pairs | `{"eval_1":"/path/to/eval_1.jsonl"}` |
| `--project` | Project name | `Project1` |
| `--num-episodes` | Number of training episodes | `1` |

View File

@ -23,7 +23,7 @@ if __name__ == "__main__":
default=None,
help="Evaluation dataset for each task, please use json format to specify the dataset for each task. \
For example: {'task1':'data_eval_task1.jsonl', 'task2':'data_eval_task2.jsonl'}, the jsonl file should be in the same format as the training dataset. \
The key is the task name, and the value is the path to the jsonl file",
The key is the task name, and the value is the path to the jsonl file, please replace sinple quotes with double quotes to conform to json format.",
)
parser.add_argument("-p", "--project", type=str, default="GRPO", help="Project name.")
parser.add_argument("-e", "--num-episodes", type=int, default=1, help="Number of episodes to train.")