Changing some things

2023-11-26 22:27:35 +00:00
parent 70fd86afa6
commit 9c84b875a3
2 changed files with 15 additions and 9 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")