From 29cc2f3a1b35006ad2e08128bd9fb49a260ba1c7 Mon Sep 17 00:00:00 2001
From: Mateusz Klimaszewski <mk.klimaszewski@gmail.com>
Date: Fri, 3 Jul 2020 15:44:23 +0200
Subject: [PATCH] Make pytorch requirements soft and silence parameters logger.
---
combo/main.py | 2 ++
setup.py | 4 ++--
2 files changed, 4 insertions(+), 2 deletions(-)
diff --git a/combo/main.py b/combo/main.py
index e2e29d4..918bb94 100644
--- a/combo/main.py
+++ b/combo/main.py
@@ -143,6 +143,8 @@ def run(_):
file.writelines(predictor.predict_instance_as_tree(tree).serialize())
else:
use_dataset_reader = ".conllu" in FLAGS.input_file.lower()
+ if FLAGS.silent:
+ logging.getLogger("allennlp.common.params").disabled = True
manager = allen_predict._PredictManager(
_get_predictor(),
FLAGS.input_file,
diff --git a/setup.py b/setup.py
index ee68a63..328d05d 100644
--- a/setup.py
+++ b/setup.py
@@ -10,8 +10,8 @@ REQUIREMENTS = [
'requests==2.23.0',
'overrides==3.0.0',
'tensorboard==2.1.0',
- 'torch==1.5.0',
- 'torchvision==0.6.0',
+ 'torch>=1.5.0',
+ 'torchvision>=0.6.0',
'tqdm==4.43.0',
'transformers==2.9.1',
'urllib3==1.24.2',
--
GitLab