diff --git a/distiller/data_loggers/collector.py b/distiller/data_loggers/collector.py index fc96aa0800072ae1c64d224b50d4bb77cf5629b6..c859924ff9345287fc0842a7f357acc5da054d98 100755 --- a/distiller/data_loggers/collector.py +++ b/distiller/data_loggers/collector.py @@ -26,6 +26,8 @@ import torch from torchnet.meter import AverageValueMeter import logging from math import sqrt +import matplotlib +matplotlib.use('Agg') import matplotlib.pyplot as plt import distiller msglogger = logging.getLogger()