diff --git a/trainers/base.py b/trainers/base.py index c4539ada85b84daefbd933f5648addc1b3c1b4b3..4b01217be775b1bfc498d4abdfd7e825403e2d18 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")