From 9f71f19bd832a148825d784711368ff49d37f70e Mon Sep 17 00:00:00 2001
From: Franck Dary <franck.dary@lis-lab.fr>
Date: Thu, 1 Jul 2021 10:59:09 +0200
Subject: [PATCH] Added script to parse train.log
---
view.py | 27 +++++++++++++++++++++++++++
1 file changed, 27 insertions(+)
create mode 100755 view.py
diff --git a/view.py b/view.py
new file mode 100755
index 0000000..97f1b2e
--- /dev/null
+++ b/view.py
@@ -0,0 +1,27 @@
+#! /usr/bin/env python3
+
+# EXAMPLE : ./view.py bin/*/*log
+
+import sys
+import subprocess
+
+cmd = 'for f in %s; do echo $f; grep "SAVED" $f | tail -1; echo ""; done'%" ".join(sys.argv[1:-1])
+p = subprocess.Popen(cmd, shell=True, stdout=subprocess.PIPE)
+
+models = []
+
+model = []
+for line in p.stdout :
+ line = line.decode("utf-8").strip()
+ if len(line) == 0 :
+ epoch = model[1].split(',')[0].split(' ')[-1]
+ name = model[0].split('/')[-2]
+ scores = model[1].split(':')[-1].replace(' SAVED', '')
+ score = sum([float(val.split('=')[-1]) for val in scores.split()])
+ models.append([score, "%s\t%s\t%s"%(epoch, scores, name)])
+ model = []
+ continue
+ model.append(line)
+
+models = sorted(models)[::-1]
+print("\n".join([m[1] for m in models]))
--
GitLab