avoid matplotlib warning

Former-commit-id: 8f9824bbaf1e8042800debbd755cc5bc5b49f437
javiribera · Sep 11, 2018 · e111d01 · e111d01
1 parent 0460f85
commit e111d01
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/object-locator/train.py b/object-locator/train.py
@@ -19,6 +19,8 @@
 from torchvision import transforms
 from torch.utils.data import DataLoader
 from sklearn import mixture
+import matplotlib
+matplotlib.use('Agg')
 import skimage.transform
 from peterpy import peter