github-clones
/
DCRNN_PyTorch
mirror of https://github.com/chnsh/DCRNN_PyTorch.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

should shuffle batches - not individual items in x_train

This commit is contained in:
Seth Ockerman 2024-07-17 13:00:44 -05:00
parent d92490b808
commit 8440002db2
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
lib/utils.py
View File

@ -186,7 +186,7 @@ def load_dataset(dataset_dir, batch_size, test_batch_size=None, **kwargs):
for category in ['train', 'val', 'test']:
data['x_' + category][..., 0] = scaler.transform(data['x_' + category][..., 0])
data['y_' + category][..., 0] = scaler.transform(data['y_' + category][..., 0])
data['train_loader'] = DataLoader(data['x_train'], data['y_train'], batch_size, shuffle=True)
data['train_loader'] = DataLoader(data['x_train'], data['y_train'], batch_size, shuffle=False)
data['val_loader'] = DataLoader(data['x_val'], data['y_val'], test_batch_size, shuffle=False)
data['test_loader'] = DataLoader(data['x_test'], data['y_test'], test_batch_size, shuffle=False)
data['scaler'] = scaler