From a4b902cdc77a938fea3210da96cfc93d5ec8b9cc Mon Sep 17 00:00:00 2001
From: Leonard Hackel <l.hackel@tu-berlin.de>
Date: Wed, 26 Apr 2023 12:32:50 +0200
Subject: [PATCH] adding dl worker param
---
pretrain_lit4rsvqa.py | 5 +++--
1 file changed, 3 insertions(+), 2 deletions(-)
diff --git a/pretrain_lit4rsvqa.py b/pretrain_lit4rsvqa.py
index 3e204fe..59a86c4 100644
--- a/pretrain_lit4rsvqa.py
+++ b/pretrain_lit4rsvqa.py
@@ -147,7 +147,8 @@ def main(
batch_size: int = 32,
seed: int = 42,
data_dir: str = None,
- test_run: bool = False
+ test_run: bool = False,
+ num_workers_dataloader: int = 4
):
if test_run:
max_img_index = 10 * batch_size
@@ -212,7 +213,7 @@ def main(
dm = BENDataModule(
data_dir=resolve_ben_data_dir(data_dir=data_dir),
img_size=(channels, img_size, img_size),
- num_workers_dataloader=4,
+ num_workers_dataloader=num_workers_dataloader,
batch_size=batch_size,
max_img_idx=max_img_index,
)
--
GitLab