From b21f94f6fc535c3fc0bdb1b75659e5ab246185c4 Mon Sep 17 00:00:00 2001
From: Tamino Huxohl <thuxohl@techfak.uni-bielefeld.de>
Date: Wed, 26 Oct 2022 12:17:04 +0200
Subject: [PATCH] loss curve plot passes on args for log parsing
---
mu_map/vis/loss_curve.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/mu_map/vis/loss_curve.py b/mu_map/vis/loss_curve.py
index c17bb84..b0ae6c7 100644
--- a/mu_map/vis/loss_curve.py
+++ b/mu_map/vis/loss_curve.py
@@ -29,8 +29,8 @@ def parse_loss(logs, loss_type):
return np.array(list(epochs)), np.array(list(losses))
-def plot_loss(logfile, loss_types, ax, normalize=False, from_epoch=0, to_epoch=None):
- logs = parse_file(logfile)
+def plot_loss(logfile, loss_types, ax, normalize=False, from_epoch=0, to_epoch=None, start_idx_message=3):
+ logs = parse_file(logfile, start_idx_message=start_idx_message)
logs = list(filter(lambda logline: logline.loglevel == "INFO", logs))
for i, loss_type in enumerate(loss_types):
--
GitLab