mirror of
https://github.com/hiyouga/LLaMA-Factory.git
synced 2025-12-18 04:40:35 +08:00
add swanlab
This commit is contained in:
@@ -305,7 +305,21 @@ class BAdamArgument:
|
||||
|
||||
|
||||
@dataclass
|
||||
class FinetuningArguments(FreezeArguments, LoraArguments, RLHFArguments, GaloreArguments, BAdamArgument):
|
||||
class SwanLabArguments:
|
||||
use_swanlab: bool = field(
|
||||
default=False,
|
||||
metadata={"help": ""},
|
||||
)
|
||||
swanlab_name: str = field(
|
||||
default="",
|
||||
metadata={},
|
||||
)
|
||||
|
||||
|
||||
@dataclass
|
||||
class FinetuningArguments(
|
||||
FreezeArguments, LoraArguments, RLHFArguments, GaloreArguments, BAdamArgument, SwanLabArguments
|
||||
):
|
||||
r"""
|
||||
Arguments pertaining to which techniques we are going to fine-tuning with.
|
||||
"""
|
||||
|
||||
Reference in New Issue
Block a user