Former-commit-id: e4123129aae59f4123d53c1f5320e3d5e09ae26d
This commit is contained in:
hiyouga
2023-11-28 20:52:28 +08:00
parent 569860d7ac
commit 670ee3934f
4 changed files with 12 additions and 6 deletions

View File

@@ -179,7 +179,7 @@ class FinetuningArguments(FreezeArguments, LoraArguments, RLHFArguments):
raise ValueError("Reward model is necessary for PPO training.")
if self.stage == "ppo" and self.reward_model_type == "lora" and self.finetuning_type != "lora":
raise ValueError("Lora reward model only supports lora training.")
raise ValueError("Freeze/Full PPO training needs `reward_model_type=full`.")
def save_to_json(self, json_path: str):
r"""Saves the content of this instance in JSON format inside `json_path`."""