Former-commit-id: b7256aa33d761280751518c20f29f9b8ea3fb025
This commit is contained in:
hiyouga
2024-04-03 14:47:59 +08:00
parent a74a7585e0
commit f6530222f7
2 changed files with 2 additions and 2 deletions

View File

@@ -120,7 +120,7 @@ def get_train_args(args: Optional[Dict[str, Any]] = None) -> _TRAIN_CLS:
if (
finetuning_args.stage == "ppo"
and training_args.report_to is not None
and training_args.report_to
and training_args.report_to[0] not in ["wandb", "tensorboard"]
):
raise ValueError("PPO only accepts wandb or tensorboard logger.")