Commit d5cd3a02 authored by Evgeny Kusmenko's avatar Evgeny Kusmenko

Merge branch 'develop' into 'master'

Fixed test

See merge request !31
parents 1c5b64d8 7c896581
Pipeline #216584 passed with stages
in 10 minutes and 50 seconds
......@@ -57,7 +57,7 @@ public class IntegrationGluonTest extends IntegrationTest {
assertTrue(Log.getFindings().isEmpty());
}
@Test
public void testShowAttendTell() {
Log.getFindings().clear();
......
......@@ -4,6 +4,8 @@ RUN apt-get update && \
apt-get install -y --no-install-recommends \
git \
libgtk2.0-dev \
python-subprocess32 \
python-tk \
wget python gcc \
build-essential cmake \
liblapack-dev libblas-dev libboost-dev libarmadillo-dev && \
......@@ -14,4 +16,4 @@ RUN git clone https://github.com/apache/incubator-mxnet.git mxnet-source && \
rm -r mxnet-source
RUN wget https://bootstrap.pypa.io/get-pip.py
RUN python get-pip.py
RUN pip install mxnet h5py opencv-python
RUN pip install mxnet h5py opencv-python matplotlib
......@@ -341,6 +341,8 @@ class CNNSupervisedTrainer_mnist_mnistClassifier_net:
if save_attention_image == "True":
import matplotlib
matplotlib.use('Agg')
import matplotlib.pyplot as plt
logging.getLogger('matplotlib').setLevel(logging.ERROR)
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment