diff --git a/train_ms.py b/train_ms.py index 3e61bf2..1a7eac6 100644 --- a/train_ms.py +++ b/train_ms.py @@ -193,8 +193,8 @@ def run(): pin_memory=True, collate_fn=collate_fn, batch_sampler=train_sampler, - persistent_workers=True, # これもメモリ消費量を減らそうとしてコメントアウト(事前ロード量?) + # persistent_workers=True, # prefetch_factor=4, ) # DataLoader config could be adjusted. eval_dataset = None diff --git a/train_ms_jp_extra.py b/train_ms_jp_extra.py index ce44ebe..9f7ab23 100644 --- a/train_ms_jp_extra.py +++ b/train_ms_jp_extra.py @@ -196,8 +196,8 @@ def run(): pin_memory=True, collate_fn=collate_fn, batch_sampler=train_sampler, - persistent_workers=True, - # これもメモリ消費量を減らそうとしてコメントアウト(事前ロード量?) + # これもメモリ消費量を減らそうとしてコメントアウト + # persistent_workers=True, # prefetch_factor=6, ) # DataLoader config could be adjusted. eval_dataset = None