From b2cde01d77e1e1207d3e8818a559b2bda0b5f7d7 Mon Sep 17 00:00:00 2001
From: litagin02 <litagin02@gmail.com>
Date: Wed, 7 Feb 2024 13:34:45 +0900
Subject: [PATCH] Fix incompatible setting

---
 train_ms.py          | 2 +-
 train_ms_jp_extra.py | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

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