mirror of
https://github.com/hiyouga/LLaMA-Factory.git
synced 2025-08-02 03:32:50 +08:00
[tracking] add swanlab_logdir param (#7219)
* feat: add swanlab_logdir param * fix Former-commit-id: a1e76af3d9cf64a6c016bb2333fc815fd4be73cf
This commit is contained in:
parent
5a29f49fb1
commit
0a43bc1960
@ -363,15 +363,15 @@ class SwanLabArguments:
|
||||
default=False,
|
||||
metadata={"help": "Whether or not to use the SwanLab (an experiment tracking and visualization tool)."},
|
||||
)
|
||||
swanlab_project: str = field(
|
||||
swanlab_project: Optional[str] = field(
|
||||
default="llamafactory",
|
||||
metadata={"help": "The project name in SwanLab."},
|
||||
)
|
||||
swanlab_workspace: str = field(
|
||||
swanlab_workspace: Optional[str] = field(
|
||||
default=None,
|
||||
metadata={"help": "The workspace name in SwanLab."},
|
||||
)
|
||||
swanlab_run_name: str = field(
|
||||
swanlab_run_name: Optional[str] = field(
|
||||
default=None,
|
||||
metadata={"help": "The experiment name in SwanLab."},
|
||||
)
|
||||
@ -379,10 +379,14 @@ class SwanLabArguments:
|
||||
default="cloud",
|
||||
metadata={"help": "The mode of SwanLab."},
|
||||
)
|
||||
swanlab_api_key: str = field(
|
||||
swanlab_api_key: Optional[str] = field(
|
||||
default=None,
|
||||
metadata={"help": "The API key for SwanLab."},
|
||||
)
|
||||
swanlab_logdir: Optional[str] = field(
|
||||
default=None,
|
||||
metadata={"help": "The log directory for SwanLab."},
|
||||
)
|
||||
|
||||
|
||||
@dataclass
|
||||
|
@ -617,6 +617,7 @@ def get_swanlab_callback(finetuning_args: "FinetuningArguments") -> "TrainerCall
|
||||
experiment_name=finetuning_args.swanlab_run_name,
|
||||
mode=finetuning_args.swanlab_mode,
|
||||
config={"Framework": "🦙LlamaFactory"},
|
||||
logdir=finetuning_args.swanlab_logdir,
|
||||
)
|
||||
return swanlab_callback
|
||||
|
||||
|
@ -1894,6 +1894,28 @@ LOCALES = {
|
||||
"info": "クラウド版またはオフライン版 SwanLab を使用します。",
|
||||
},
|
||||
},
|
||||
"swanlab_logdir": {
|
||||
"en": {
|
||||
"label": "SwanLab log directory",
|
||||
"info": "The log directory for SwanLab.",
|
||||
},
|
||||
"ru": {
|
||||
"label": "SwanLab 로그 디렉토리",
|
||||
"info": "SwanLab의 로그 디렉토리.",
|
||||
},
|
||||
"zh": {
|
||||
"label": "SwanLab 日志目录",
|
||||
"info": "SwanLab 的日志目录。",
|
||||
},
|
||||
"ko": {
|
||||
"label": "SwanLab 로그 디렉토리",
|
||||
"info": "SwanLab의 로그 디렉토리.",
|
||||
},
|
||||
"ja": {
|
||||
"label": "SwanLab ログ ディレクトリ",
|
||||
"info": "SwanLab のログ ディレクトリ。",
|
||||
},
|
||||
},
|
||||
"cmd_preview_btn": {
|
||||
"en": {
|
||||
"value": "Preview command",
|
||||
|
Loading…
x
Reference in New Issue
Block a user