Add 'matplotlib.use('Agg')'; update output path

2026-06-10 16:14:07 +02:00
parent 2ef6f79480
commit fd76272157
6 changed files with 31 additions and 9 deletions
@@ -1,5 +1,8 @@
 import sys
 sys.path.append('./src')
+import matplotlib
+matplotlib.use('Agg')
+
 from pathlib import Path
 from omegaconf import OmegaConf
 import torch