Former-commit-id: 0bd884feb11736d0ab24ca19885151cb47d9dcd3
This commit is contained in:
hiyouga 2023-11-10 14:38:18 +08:00
parent 178b85ff9a
commit f0766a2ab0

View File

@ -54,11 +54,11 @@ class ModelArguments:
default=False,
metadata={"help": "Enable shift short attention (S^2-Attn) proposed by LongLoRA."}
)
reward_model: Optional[str] = field(
reward_model: Optional[str] = field( # TODO: move it to FinetuningArguments
default=None,
metadata={"help": "Path to the directory containing the checkpoints of the reward model."}
)
plot_loss: Optional[bool] = field(
plot_loss: Optional[bool] = field( # TODO: move it to FinetuningArguments
default=False,
metadata={"help": "Whether to plot the training loss after fine-tuning or not."}
)