Added training file

2023-11-27 11:53:44 +00:00
parent 74c2c07b68
commit 5e87165dbb
2 changed files with 70 additions and 1 deletions
--- a/src/utils/clearml.py
+++ b/src/utils/clearml.py
@@ -5,7 +5,7 @@ class ClearMLHelper:
        self.project_name = project_name

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