From 3e9a11fa36be299dd3a9d4629687f0d5ff32ccbb Mon Sep 17 00:00:00 2001 From: Frozenmad Date: Thu, 29 Apr 2021 11:25:17 +0000 Subject: [PATCH] change the default num_workers to 0 --- autogl/module/train/graph_classification_full.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/autogl/module/train/graph_classification_full.py b/autogl/module/train/graph_classification_full.py index f6b32ae..1f4bb4a 100644 --- a/autogl/module/train/graph_classification_full.py +++ b/autogl/module/train/graph_classification_full.py @@ -98,7 +98,7 @@ class GraphClassificationFullTrainer(BaseGraphClassificationTrainer): self.lr = lr if lr is not None else 1e-4 self.max_epoch = max_epoch if max_epoch is not None else 100 self.batch_size = batch_size if batch_size is not None else 64 - self.num_workers = num_workers if num_workers is not None else 4 + self.num_workers = num_workers if num_workers is not None else 0 if self.num_workers > 0: mp.set_start_method("fork", force=True) self.early_stopping_round = (