Browse Source

change the default num_workers to 0

tags/v0.3.1
Frozenmad 5 years ago
parent
commit
3e9a11fa36
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      autogl/module/train/graph_classification_full.py

+ 1
- 1
autogl/module/train/graph_classification_full.py View File

@@ -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 = (


Loading…
Cancel
Save