VictorMylle/Thesis
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Changes to allow remote execution

Browse Source
This commit is contained in:
Victor Mylle
2023-11-25 23:36:57 +00:00
parent 300f268286
commit a8db70e86d
7 changed files with 4500091 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/utils/clearml.py
View File

@@ -6,5 +6,5 @@ class ClearMLHelper:
def get_task(self, task_name: str = "Model Training"):
task = Task.init(project_name=self.project_name, task_name=task_name, continue_last_task=False)
task.set_base_docker(f"pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime")
task.set_base_docker(f"docker.io/clearml/pytorch-cuda-gcc:2.0.0-cuda11.7-cudnn8-runtime --env GIT_SSL_NO_VERIFY=true --env CLEARML_AGENT_GIT_USER=VictorMylle --env CLEARML_AGENT_GIT_PASS=Voetballer1" )
return task