From e9b8d1662b0755e06703e0934b9da4f0d9417b8a Mon Sep 17 00:00:00 2001 From: Kai Su <67154044+kai3316@users.noreply.github.com> Date: Fri, 30 Jul 2021 18:16:49 +0900 Subject: [PATCH 1/3] Update requirements.txt --- requirements.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/requirements.txt b/requirements.txt index fea6b09..7f31f70 100644 --- a/requirements.txt +++ b/requirements.txt @@ -7,7 +7,7 @@ torch==1.0.1.post2 torchvision==0.2.2 prettytable==0.7.2 numpy==1.16.2 -tqdm==4.29.1 +tqdm==4.19.9 coloredlogs==10.0 scipy==1.2.1 setproctitle==1.1.10 From 4ba1e1f14ca7acd4c20fa723e6cb4b290baa2b3e Mon Sep 17 00:00:00 2001 From: Kai Su <67154044+kai3316@users.noreply.github.com> Date: Fri, 30 Jul 2021 20:42:08 +0900 Subject: [PATCH 2/3] Update requirements.txt --- requirements.txt | 1 + 1 file changed, 1 insertion(+) diff --git a/requirements.txt b/requirements.txt index 7f31f70..d590f33 100644 --- a/requirements.txt +++ b/requirements.txt @@ -15,3 +15,4 @@ tensorflow_gpu==1.9.0 matplotlib==2.0.2 visdom==0.1.8.8 Cython==0.29.6 +scikit-learn==0.22.1 From 41abb5e3a0f975efcd43c3057be68bba13cc660d Mon Sep 17 00:00:00 2001 From: Kai Su <67154044+kai3316@users.noreply.github.com> Date: Fri, 30 Jul 2021 21:06:33 +0900 Subject: [PATCH 3/3] Update visualize.py change for different environments. --- src/m_utils/visualize.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/m_utils/visualize.py b/src/m_utils/visualize.py index 3bd89a4..f6f4e2d 100644 --- a/src/m_utils/visualize.py +++ b/src/m_utils/visualize.py @@ -1,4 +1,8 @@ - +import os +import matplotlib as mpl +if os.environ.get('DISPLAY','') == '': + print('no display found. Using non-interactive Agg backend') + mpl.use('Agg') import matplotlib.pyplot as plt import seaborn as sns from PIL import Image