From b95954f931435f40f55e8974108973f0d5ab9b5b Mon Sep 17 00:00:00 2001 From: Adithya U R Date: Mon, 1 Aug 2022 17:19:44 +0200 Subject: [PATCH] To extract the correct label name --- tools/analyze_logs.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/analyze_logs.py b/tools/analyze_logs.py index 2810c98f..de425d2f 100644 --- a/tools/analyze_logs.py +++ b/tools/analyze_logs.py @@ -79,7 +79,7 @@ def plot_curve(log_dicts, args): ys = np.concatenate(ys) plt.xlabel('iter') plt.plot( - xs, ys, label=legend[i * num_metrics + j], linewidth=0.5) + xs, ys, label=args.keys[j], linewidth=0.5) plt.legend() if args.title is not None: plt.title(args.title)