423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-08-04 20:52:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

add todo

Browse Source 
Former-commit-id: a0c31c68c4909637b86c90c319c321fd887c4910
This commit is contained in:
hiyouga 2023-11-10 14:38:18 +08:00
parent 0fbaa42752
commit 55e097aaac
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/llmtuner/hparams/model_args.py
View File

@ -54,11 +54,11 @@ class ModelArguments:
default=False, default=False,
metadata={"help": "Enable shift short attention (S^2-Attn) proposed by LongLoRA."} 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, default=None,
metadata={"help": "Path to the directory containing the checkpoints of the reward model."} 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, default=False,
metadata={"help": "Whether to plot the training loss after fine-tuning or not."} metadata={"help": "Whether to plot the training loss after fine-tuning or not."}
) )