From 8b5344ee2dc551dbe673020a33021e7f84f305f1 Mon Sep 17 00:00:00 2001
From: AlexeyAB <alexeyab84@gmail.com>
Date: Mon, 14 May 2018 10:34:40 +0000
Subject: [PATCH] Added BFLOPs output for network configurations
---
scripts/log_parser/log_parser.py | 4 +++-
1 files changed, 3 insertions(+), 1 deletions(-)
diff --git a/scripts/log_parser/log_parser.py b/scripts/log_parser/log_parser.py
index 2804c02..631e549 100644
--- a/scripts/log_parser/log_parser.py
+++ b/scripts/log_parser/log_parser.py
@@ -10,8 +10,8 @@
import platform
import re
import sys
-
import matplotlib.pyplot as plt
+
from matplotlib.ticker import MultipleLocator, FormatStrFormatter
@@ -98,6 +98,8 @@
plt.savefig(save_path, dpi=300)
if args.show_plot:
plt.show()
+ else:
+ plt.switch_backend('agg')
if __name__ == "__main__":
--
Gitblit v1.10.0