Skip to content

Commit a251459

Browse files
authored
update shuffle trainer.py
1 parent 1bbba45 commit a251459

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

pina/trainer.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,7 @@ def __init__(
5656
:param pin_memory: Whether to use pinned memory for faster data
5757
transfer to GPU. Default False.
5858
:type pin_memory: bool
59-
:param shuffle: Whether to shuffle the data for training. Default False.
59+
:param shuffle: Whether to shuffle the data for training. Default True.
6060
:type pin_memory: bool
6161
6262
:Keyword Arguments:
@@ -85,7 +85,7 @@ def __init__(
8585
if shuffle is not None:
8686
check_consistency(shuffle, bool)
8787
else:
88-
shuffle = False
88+
shuffle = True
8989
if train_size + test_size + val_size + predict_size > 1:
9090
raise ValueError(
9191
"train_size, test_size, val_size and predict_size "

0 commit comments

Comments
 (0)