From 0d609c80449e277308f71673059444169593c6b0 Mon Sep 17 00:00:00 2001
From: Pierre LOTTE <pierrelotte.dev@gmail.com>
Date: Wed, 18 Sep 2024 10:33:43 +0200
Subject: [PATCH] Save time only during training
---
trainers/base.py | 5 +++--
1 file changed, 3 insertions(+), 2 deletions(-)
diff --git a/trainers/base.py b/trainers/base.py
index c4539ad..4b01217 100644
--- a/trainers/base.py
+++ b/trainers/base.py
@@ -65,5 +65,6 @@ class BaseTrainer():
)
duration = time() - start_time
- with open(f"{path}/time.csv", "a", encoding="utf-8") as f:
- f.write(f"{self.algorithm},{file_name},{duration}\n")
+ if task == "execute":
+ with open(f"{path}/time.csv", "a", encoding="utf-8") as f:
+ f.write(f"{self.algorithm},{file_name},{duration}\n")
--
GitLab