fix: by hiyouga suggestion

This commit is contained in:
ZeYi Lin
2024-12-20 16:43:03 +08:00
parent 5f6dafd70e
commit 3a7ea2048a
7 changed files with 23 additions and 8 deletions

View File

@@ -308,10 +308,10 @@ class BAdamArgument:
class SwanLabArguments:
use_swanlab: bool = field(
default=False,
metadata={"help": "Whether or not to use the SwanLab (an experiment tracking and visualization tools)."},
metadata={"help": "Whether or not to use the SwanLab (an experiment tracking and visualization tool)."},
)
swanlab_project: str = field(
default=None,
default="LLaMA Factory",
metadata={"help": "The project name in SwanLab."},
)
swanlab_workspace: str = field(