folivo/PINA
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update shuffle trainer.py

Browse Source
This commit is contained in:
Dario Coscia
2025-03-05 12:32:43 +01:00
committed by Nicola Demo
parent ab6ca78d85
commit 4cb0987714
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
pina/trainer.py
View File

@@ -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 "