Skip to main content

MMCV v1.x

info

ClearMLLoggerHook 由 mmcv =>1.5.1 和 <=1.7.0 支持。

tip

如果您尚未使用ClearML，请参阅入门指南以获取设置说明。

MMCV 是由 OpenMMLab 开发的计算机视觉框架。你可以使用 mmcv 包的 ClearMLLoggerHook 类将 ClearML 集成到你的代码中。这个类用于创建一个 ClearML 任务并自动记录指标。

例如，以下代码设置了定期将指标记录到ClearML的配置，然后将ClearML钩子注册到runner，该runner在mmcv中管理训练：

log_config = dict(
    interval=100,
    hooks=[
        dict(
            type='ClearMLLoggerHook',
            init_kwargs=dict(
                project_name='examples',
                task_name='OpenMMLab cifar10',
                output_uri=True
            )
        ),
    ]
)

# register hooks to runner and those hooks will be invoked automatically
runner.register_training_hooks(
    lr_config=lr_config,
    optimizer_config=optimizer_config,
    checkpoint_config=checkpoint_config,
    log_config=log_config # ClearMLLogger hook
)

init_kwargs 字典可以包含来自 Task.init() 的任何参数。

这将创建一个ClearML任务 OpenMMLab cifar10 在 examples 项目中。 您可以在WebApp的实验的标量标签中查看捕获的指标。

OpenMMLab 标量

查看MMCV代码示例 这里.