update callback.py

bubbliiiing · Mar 26, 2022 · 956d6cf · 956d6cf
1 parent aec8b77
commit 956d6cf
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/.gitignore b/.gitignore
@@ -4,6 +4,9 @@ miou_out/
 VOCdevkit/
 datasets/
 Medical_Datasets/
+lfw/
+logs/
+model_data/
 
 # Byte-compiled / optimized / DLL files
 __pycache__/

diff --git a/utils/callbacks.py b/utils/callbacks.py
@@ -1,5 +1,7 @@
 import os
 
+import matplotlib
+matplotlib.use('Agg')
 import scipy.signal
 from matplotlib import pyplot as plt