Commit baee5080 authored by Evgeny Kusmenko's avatar Evgeny Kusmenko

Merge branch 'develop' into 'master'

fixed a small bug when using matplotlib

See merge request !24
parents 075c2f28 bcb8841a
Pipeline #216537 passed with stages
in 7 minutes and 19 seconds
if save_attention_image == "True":
import matplotlib
matplotlib.use('Agg')
import matplotlib.pyplot as plt
logging.getLogger('matplotlib').setLevel(logging.ERROR)
......
......@@ -341,6 +341,8 @@ class CNNSupervisedTrainer_Alexnet:
if save_attention_image == "True":
import matplotlib
matplotlib.use('Agg')
import matplotlib.pyplot as plt
logging.getLogger('matplotlib').setLevel(logging.ERROR)
......
......@@ -341,6 +341,8 @@ class CNNSupervisedTrainer_CifarClassifierNetwork:
if save_attention_image == "True":
import matplotlib
matplotlib.use('Agg')
import matplotlib.pyplot as plt
logging.getLogger('matplotlib').setLevel(logging.ERROR)
......
......@@ -341,6 +341,8 @@ class CNNSupervisedTrainer_VGG16:
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