init🎉:
This commit is contained in:
11
train/configs/finetune_arguments.py
Normal file
11
train/configs/finetune_arguments.py
Normal file
@ -0,0 +1,11 @@
|
||||
from dataclasses import dataclass, field
|
||||
|
||||
|
||||
### 定义一些配置信息
|
||||
@dataclass
|
||||
class FinetuneArguments:
|
||||
model_name: str = field()
|
||||
data_path: str = field()
|
||||
train_size: int = field(default=-1)
|
||||
test_size: int = field(default=100)
|
||||
max_len: int = field(default=1024)
|
||||
Reference in New Issue
Block a user