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

Fix incompatible setting

Browse Source
This commit is contained in:
litagin02
2024-02-07 13:34:45 +09:00
parent 8898e5f600
commit b2cde01d77
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
train_ms.py
View File

@@ -193,8 +193,8 @@ def run():
pin_memory=True, pin_memory=True,
collate_fn=collate_fn, collate_fn=collate_fn,
batch_sampler=train_sampler, batch_sampler=train_sampler,
persistent_workers=True,
# これもメモリ消費量を減らそうとしてコメントアウト（事前ロード量？） # これもメモリ消費量を減らそうとしてコメントアウト（事前ロード量？）
# persistent_workers=True,
# prefetch_factor=4, # prefetch_factor=4,
) # DataLoader config could be adjusted. ) # DataLoader config could be adjusted.
eval_dataset = None eval_dataset = None

4
train_ms_jp_extra.py
View File

@@ -196,8 +196,8 @@ def run():
pin_memory=True, pin_memory=True,
collate_fn=collate_fn, collate_fn=collate_fn,
batch_sampler=train_sampler, batch_sampler=train_sampler,
persistent_workers=True, # これもメモリ消費量を減らそうとしてコメントアウト
# これもメモリ消費量を減らそうとしてコメントアウト（事前ロード量？） # persistent_workers=True,
# prefetch_factor=6, # prefetch_factor=6,
) # DataLoader config could be adjusted. ) # DataLoader config could be adjusted.
eval_dataset = None eval_dataset = None