Change datamodules' default values

Lightning-Universe · Aug 12, 2021 · c6adbae · c6adbae
1 parent 4eebe3f
commit c6adbae
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/pl_bolts/datamodules/binary_emnist_datamodule.py b/pl_bolts/datamodules/binary_emnist_datamodule.py
@@ -29,12 +29,12 @@ def __init__(
         data_dir: Optional[str] = None,
         split: str = 'mnist',
         val_split: Union[int, float] = 0.2,
-        num_workers: int = 16,
+        num_workers: int = 0,
         normalize: bool = False,
         batch_size: int = 32,
         seed: int = 42,
-        shuffle: bool = False,
-        pin_memory: bool = False,
+        shuffle: bool = True,
+        pin_memory: bool = True,
         drop_last: bool = False,
         strict_val_split: bool = False,
         *args: Any,

diff --git a/pl_bolts/datamodules/emnist_datamodule.py b/pl_bolts/datamodules/emnist_datamodule.py
@@ -105,12 +105,12 @@ def __init__(
         data_dir: Optional[str] = None,
         split: str = 'mnist',
         val_split: Union[int, float] = 0.2,
-        num_workers: int = 16,
+        num_workers: int = 0,
         normalize: bool = False,
         batch_size: int = 32,
         seed: int = 42,
-        shuffle: bool = False,
-        pin_memory: bool = False,
+        shuffle: bool = True,
+        pin_memory: bool = True,
         drop_last: bool = False,
         strict_val_split: bool = False,
         *args: Any,