Unverified 提交 1ce464f6 authored 作者: Victor Sonck's avatar Victor Sonck 提交者: GitHub

Add docker info for ClearML remote execution (#10142)

* Add docker info for ClearML remote execution * add additional clearml options to handle different python versions * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ciCo-authored-by: 's avatarGlenn Jocher <glenn.jocher@ultralytics.com> Co-authored-by: 's avatarpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
上级 185d475d
...@@ -97,6 +97,11 @@ class ClearmlLogger: ...@@ -97,6 +97,11 @@ class ClearmlLogger:
# will have to be added manually! # will have to be added manually!
self.task.connect(hyp, name='Hyperparameters') self.task.connect(hyp, name='Hyperparameters')
# Make sure the code is easily remotely runnable by setting the docker image to use by the remote agent
self.task.set_base_docker("ultralytics/yolov5:latest",
docker_arguments='--ipc=host -e="CLEARML_AGENT_SKIP_PYTHON_ENV_INSTALL=1"',
docker_setup_bash_script='pip install clearml')
# Get ClearML Dataset Version if requested # Get ClearML Dataset Version if requested
if opt.data.startswith('clearml://'): if opt.data.startswith('clearml://'):
# data_dict should have the following keys: # data_dict should have the following keys:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论