From ec660e3131f8898cfe6570a32cda2cff7543e0e8 Mon Sep 17 00:00:00 2001 From: Matthew Honnibal Date: Tue, 1 Sep 2020 00:41:38 +0200 Subject: [PATCH] Fix use_pytorch_for_gpu_memory --- spacy/cli/train.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/spacy/cli/train.py b/spacy/cli/train.py index b688dd384..7525b9669 100644 --- a/spacy/cli/train.py +++ b/spacy/cli/train.py @@ -77,6 +77,9 @@ def train( ) if config.get("training", {}).get("seed") is not None: fix_random_seed(config["training"]["seed"]) + if config.get("system", {}).get("use_pytorch_for_gpu_memory"): + # It feels kind of weird to not have a default for this. + use_pytorch_for_gpu_memory() # Use original config here before it's resolved to functions sourced_components = get_sourced_components(config) with show_validation_error(config_path): @@ -85,9 +88,6 @@ def train( util.load_vectors_into_model(nlp, config["training"]["vectors"]) verify_config(nlp) raw_text, tag_map, morph_rules, weights_data = load_from_paths(config) - if config.get("system", {}).get("use_pytorch_for_gpu_memory"): - # It feels kind of weird to not have a default for this. - use_pytorch_for_gpu_memory() T_cfg = config["training"] optimizer = T_cfg["optimizer"] train_corpus = T_cfg["train_corpus"]