From 8332c1a6d956748db288d8edb7797ca38a5ea60e Mon Sep 17 00:00:00 2001
From: Simon Kirsten <1972314+skirsten@users.noreply.github.com>
Date: Wed, 26 Oct 2022 16:24:48 +0100
Subject: [PATCH] Enable multi-process DataLoader for dreambooth (#950)

---
 examples/dreambooth/train_dreambooth.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/examples/dreambooth/train_dreambooth.py b/examples/dreambooth/train_dreambooth.py
index c28bcbd2..6dbba9af 100644
--- a/examples/dreambooth/train_dreambooth.py
+++ b/examples/dreambooth/train_dreambooth.py
@@ -502,7 +502,7 @@ def main():
         return batch
 
     train_dataloader = torch.utils.data.DataLoader(
-        train_dataset, batch_size=args.train_batch_size, shuffle=True, collate_fn=collate_fn
+        train_dataset, batch_size=args.train_batch_size, shuffle=True, collate_fn=collate_fn, num_workers=1
     )
 
     # Scheduler and math around the number of training steps.