mirror of
https://github.com/hiyouga/LLaMA-Factory.git
synced 2025-08-05 05:02:50 +08:00
14 lines
361 B
Python
14 lines
361 B
Python
from typing import Literal, Optional
|
|
from dataclasses import dataclass, field
|
|
|
|
|
|
@dataclass
|
|
class GeneralArguments:
|
|
"""
|
|
Arguments pertaining to which stage we are going to perform.
|
|
"""
|
|
stage: Optional[Literal["pt", "sft", "rm", "ppo"]] = field(
|
|
default="sft",
|
|
metadata={"help": "Which stage will be performed in training."}
|
|
)
|