From 71cc43e65cd47c6704d20c99006a3e78feb2400d Mon Sep 17 00:00:00 2001 From: mrq Date: Sun, 26 Feb 2023 14:56:26 +0000 Subject: [PATCH] added a flag (thanks gannybal) --- codes/data/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/codes/data/__init__.py b/codes/data/__init__.py index ac426ba1..81a915ad 100644 --- a/codes/data/__init__.py +++ b/codes/data/__init__.py @@ -20,7 +20,7 @@ def create_dataloader(dataset, dataset_opt, opt=None, sampler=None, collate_fn=N batch_size = dataset_opt['batch_size'] return torch.utils.data.DataLoader(dataset, batch_size=batch_size, shuffle=shuffle, num_workers=num_workers, sampler=sampler, drop_last=True, - pin_memory=pin_memory, collate_fn=collate_fn) + pin_memory=pin_memory, collate_fn=collate_fn, persistent_workers=True) else: batch_size = dataset_opt['batch_size'] or 1 return torch.utils.data.DataLoader(dataset, batch_size=batch_size, shuffle=False, num_workers=0,