Commit bcb8841a authored by Christian Fuß's avatar Christian Fuß

fixed a small bug when using matplotlib

parent aff4db2f
Pipeline #215859 passed with stages
in 4 minutes and 42 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