tuna2134/sbv2-v2
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Update config.py

Browse Source
This commit is contained in:
Stardust·减
2023-11-25 17:31:45 +08:00
committed by GitHub
parent 8e28d5e5c5
commit 241c6e4af3
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
config.py
View File

@@ -122,6 +122,7 @@ class Train_ms_config:
model: str, model: str,
num_workers: int, num_workers: int,
spec_cache: bool, spec_cache: bool,
keep_ckpts: int,
): ):
self.env = env # 需要加载的环境变量 self.env = env # 需要加载的环境变量
self.base = base # 底模配置 self.base = base # 底模配置
@@ -129,6 +130,7 @@ class Train_ms_config:
self.config_path = config_path # 配置文件路径 self.config_path = config_path # 配置文件路径
self.num_workers = num_workers # worker数量 self.num_workers = num_workers # worker数量
self.spec_cache = spec_cache # 是否启用spec缓存 self.spec_cache = spec_cache # 是否启用spec缓存
self.keep_ckpts = keep_ckpts # ckpt数量
@classmethod @classmethod
def from_dict(cls, dataset_path: str, data: Dict[str, any]): def from_dict(cls, dataset_path: str, data: Dict[str, any]):