Added time as input feature

2023-11-26 18:43:03 +00:00
parent a2c9427d16
commit 2f40f41786
9 changed files with 168 additions and 138 deletions
--- a/src/utils/clearml.py
+++ b/src/utils/clearml.py
@@ -5,6 +5,7 @@ class ClearMLHelper:
        self.project_name = project_name

    def get_task(self, task_name: str = "Model Training"):
+        Task.add_requirements("requirements.txt")
        Task.ignore_requirements("torch")
        Task.ignore_requirements("torchvision")
        Task.ignore_requirements("tensorboard")