Skip to content

Commit 2a3260e

Browse files
fix checks
1 parent da1fa68 commit 2a3260e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pina/trainer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -330,7 +330,7 @@ def _check_consistency_and_set_defaults(
330330
else:
331331
pin_memory = False
332332
if num_workers is not None:
333-
check_consistency(pin_memory, int)
333+
check_consistency(num_workers, int)
334334
else:
335335
num_workers = 0
336336
if shuffle is not None:

0 commit comments

Comments
 (0)