Update finetuning_args.py
This commit is contained in:
1 parent
7f9770b2c6
commit
b24635d22b
1 file changed
+4
@@ -74,6 +74,10 @@ class RLHFArguments:
|
|||||||
default=None,
|
default=None,
|
||||||
metadata={"help": "Log with either 'wandb' or 'tensorboard' in PPO training."}
|
metadata={"help": "Log with either 'wandb' or 'tensorboard' in PPO training."}
|
||||||
)
|
)
|
||||||
|
ppo_epochs: Optional[int] = field(
|
||||||
|
default=4,
|
||||||
|
metadata={"help": "Number of optimisation epochs per batch of samples"},
|
||||||
|
)
|
||||||
ppo_score_norm: Optional[bool] = field(
|
ppo_score_norm: Optional[bool] = field(
|
||||||
default=False,
|
default=False,
|
||||||
metadata={"help": "Use score normalization in PPO training."}
|
metadata={"help": "Use score normalization in PPO training."}
|
||||||
|
|||||||
Reference in new issue
Block a user