Spaces:

chenyangqi
/

FateZero

Runtime error

chenyangqi commited on Mar 28, 2023

Commit

590b13e

•

1 Parent(s): a24623d

reduce worker

Files changed (2) hide show

FateZero/test_fatezero.py CHANGED Viewed

@@ -150,7 +150,7 @@ def test(
         train_dataset,
         batch_size=train_batch_size,
         shuffle=True,
-        num_workers=4,
         collate_fn=collate_fn,
     )
     train_sample_save_path = os.path.join(logdir, "train_samples.gif")

         train_dataset,
         batch_size=train_batch_size,
         shuffle=True,
+        num_workers=0,
         collate_fn=collate_fn,
     )
     train_sample_save_path = os.path.join(logdir, "train_samples.gif")

FateZero/train_tune_a_video.py CHANGED Viewed

@@ -240,7 +240,7 @@ def train(
         train_dataset,
         batch_size=train_batch_size,
         shuffle=True,
-        num_workers=16,
         collate_fn=collate_fn,
     )

         train_dataset,
         batch_size=train_batch_size,
         shuffle=True,
+        num_workers=0,
         collate_fn=collate_fn,
     )