Fix incompatible setting

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

View File

@@ -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