def test_data_loader_batch(batch_size_nuevo):
_, _, data_test = load_data_func()
test_loader = torch.utils.data.DataLoader(data_test,
collate_fn=sort_batch,
batch_size=batch_size_nuevo,
shuffle=False,
num_workers=2,
pin_memory=True)
return test_loader
def all_data_loader():
data_train, data_valid, data_test = load_data_func()
train_loader = torch.utils.data.DataLoader(data_train,
collate_fn=sort_batch,
batch_size=BATCH_SIZE,
shuffle=True,
num_workers=2,
pin_memory=True)
valid_loader = torch.utils.data.DataLoader(data_valid,
collate_fn=sort_batch,
batch_size=BATCH_SIZE,
shuffle=False,
num_workers=2,
pin_memory=True)
test_loader = torch.utils.data.DataLoader(data_test,
collate_fn=sort_batch,
batch_size=BATCH_SIZE,
shuffle=False,
num_workers=2,
pin_memory=True)
return train_loader, valid_loader, test_loader