diff --git a/pina/trainer.py b/pina/trainer.py index 8e1d95110..e92928d1e 100644 --- a/pina/trainer.py +++ b/pina/trainer.py @@ -330,7 +330,7 @@ def _check_consistency_and_set_defaults( else: pin_memory = False if num_workers is not None: - check_consistency(pin_memory, int) + check_consistency(num_workers, int) else: num_workers = 0 if shuffle is not None: