Skip to content

Commit 00630fb

Browse files
committed
Do not pin memory
Let's see if this reduces memory utilization
1 parent 7cd0ff4 commit 00630fb

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

src/data/discotube_text_audio.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -214,7 +214,7 @@ def train_dataloader(self):
214214
self.dataset_train,
215215
batch_size=self.batch_size,
216216
num_workers=self.num_workers,
217-
pin_memory=True,
217+
pin_memory=False,
218218
collate_fn=self.collate_fn,
219219
)
220220

@@ -223,6 +223,6 @@ def val_dataloader(self):
223223
self.dataset_val,
224224
batch_size=self.batch_size,
225225
num_workers=self.num_workers,
226-
pin_memory=True,
226+
pin_memory=False,
227227
collate_fn=self.collate_fn,
228228
)

0 commit comments

Comments
 (0)