From 4cb098771415545a746a7ff399700a51765b3634 Mon Sep 17 00:00:00 2001 From: Dario Coscia <93731561+dario-coscia@users.noreply.github.com> Date: Wed, 5 Mar 2025 12:32:43 +0100 Subject: [PATCH] update shuffle trainer.py --- pina/trainer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pina/trainer.py b/pina/trainer.py index f0175e9..d1bd93d 100644 --- a/pina/trainer.py +++ b/pina/trainer.py @@ -56,7 +56,7 @@ class Trainer(lightning.pytorch.Trainer): :param pin_memory: Whether to use pinned memory for faster data transfer to GPU. Default False. :type pin_memory: bool - :param shuffle: Whether to shuffle the data for training. Default False. + :param shuffle: Whether to shuffle the data for training. Default True. :type pin_memory: bool :Keyword Arguments: @@ -85,7 +85,7 @@ class Trainer(lightning.pytorch.Trainer): if shuffle is not None: check_consistency(shuffle, bool) else: - shuffle = False + shuffle = True if train_size + test_size + val_size + predict_size > 1: raise ValueError( "train_size, test_size, val_size and predict_size "