diff --git a/pl_bolts/datamodules/binary_emnist_datamodule.py b/pl_bolts/datamodules/binary_emnist_datamodule.py index a4fcc30ecd..9dbd6d2040 100644 --- 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 index a76c979545..2288155953 100644 --- 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,