Update main.py

Easycelsius · Jun 4, 2021 · cd6a20a · cd6a20a
1 parent 82856b9
commit cd6a20a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/bin/main.py b/bin/main.py
@@ -77,7 +77,7 @@ def train(config: DictConfig) -> nn.DataParallel:
     torch.manual_seed(config.train.seed)
     torch.cuda.manual_seed_all(config.train.seed)
     device = check_envirionment(config.train.use_cuda)
-    if config.train.num_threads and int(config.train.num_threads) > 0:
+    if hasattr(config.train, "num_threads") and int(config.train.num_threads) > 0:
         torch.set_num_threads(config.train.num_threads)
 
     vocab = KsponSpeechVocabulary(