We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 7cd0ff4 commit 00630fbCopy full SHA for 00630fb
1 file changed
src/data/discotube_text_audio.py
@@ -214,7 +214,7 @@ def train_dataloader(self):
214
self.dataset_train,
215
batch_size=self.batch_size,
216
num_workers=self.num_workers,
217
- pin_memory=True,
+ pin_memory=False,
218
collate_fn=self.collate_fn,
219
)
220
@@ -223,6 +223,6 @@ def val_dataloader(self):
223
self.dataset_val,
224
225
226
227
228
0 commit comments