From bf8de726059765942861a71e6d567d8b7069a277 Mon Sep 17 00:00:00 2001 From: lwwang1995 Date: Sat, 5 Dec 2020 22:39:30 +0800 Subject: [PATCH] Update test_dataset --- tests/test_dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_dataset.py b/tests/test_dataset.py index 3c55db54b..5c0c9d843 100755 --- a/tests/test_dataset.py +++ b/tests/test_dataset.py @@ -62,7 +62,7 @@ class TestDataset(TestAutoData): ) tsds_train = tsdh.prepare("train") # Test the correctness tsds = tsdh.prepare("valid") # prepare a dataset with is friendly to converting tabular data to time-series - train_loader = DataLoader(tsds_train, batch_size=800, shuffle=True) + train_loader = DataLoader(tsds_train, batch_size=800, shuffle=True, num_workers=10) for data in train_loader: now = time.localtime() print(time.strftime("%Y-%m-%d-%H_%M_%S", now))